D. Multiple Testcases(贪心、优先队列)

题目传送门

题意: 给你一个数组,里面有n个数,给你一个k,再给你k个c[i],你将对原数组进行分组,分组的要求是分出来的每个组中,大于等于i的数不超过c[i]个,求最少可以分成多少组,并输出每一组中的数据,多个答案则任意打印。

思路: 我们要先安排大的数, 因为大的数要比小的数更难安排,我们从大到小去安排,记录第i组里放了j个数,把它放到优先队列里,j小优先,j相同的话i小优先。我们遍历原数组的时候,每次从优先队列中拿出一个,看看这个C【i】能否放进去,能放进去则放,不能放的话我们只能新开一个分组,因为我们从优先队列中拿出来的已经是最小的了,我们不可能再找到另一个可以放a[i]的分组。

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
#define int long long
#define vi vector<int>
#define mii map<int,int>
#define pii pair<int,int>
#define ull unsigned long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){
int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
return x*f;}
using namespace std;
const int N=2e5+5;
const int inf=0x7fffffff;
const int mod=1e9+7;
const double eps=1e-6;
int a[N],c[N];
vi v[N];
bool cmp(int a,int b)
{
    return a>b;
}
priority_queue<pii,vector<pii>,greater<pii> >q;
signed main()
{
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    sort(a+1,a+n+1,cmp);
    for(int i=1;i<=k;i++)
        cin>>c[i];
    int cnt=0;
    q.push(mp(0,0));
    for(int i=1;i<=n;i++)
    {
        auto x=q.top();q.pop();
        if(x.fi<c[a[i]])//可以放在第x.second个分组中
        {
            v[x.se].pb(a[i]);
            q.push(mp(x.fi+1,x.se));//新加了一个数,所以x.fi要+1
        }
        else
        {
            cnt++;
            v[cnt].pb(a[i]);//新加一个分组
            q.push(x);//原来这个要回到队列中
            q.push(mp(1,cnt));//新增加的分组
        }
    }
    cout<<cnt+1<<endl;
    for(int i=0;i<=cnt;i++)
    {
        cout<<v[i].size()<<' ';
        for(auto j:v[i])
            cout<<j<<' ';
        cout<<endl;
    }
}

原创文章 144 获赞 13 访问量 8680

猜你喜欢

转载自blog.csdn.net/Joker_He/article/details/105838578