CF1342 D - Multiple Testcases

D - Multiple Testcases

Question

給定 $n$ 個數 $m[i]$，每個$m[i]$都在 $[1,k]$的範圍內
再給定$k$個數$c[i]$
要求將所有的$m[i]$進行分組
$c[i]$表示每組中大於等於$i$的數不超過$c[i]$個
問最少能分幾組，並輸出分組方案。

Solution

分組記入不同$m[i]$的貢獻值，求一個後綴和$cnt[i]$表示$\geq i$的數有多少個。
首先考慮最小分組$ans$，$ans=max~ \lceil \frac {cnt[i]}{c[i]} \rceil$
接着分組組數確定了，我們只要構造每組都滿足題意儘可能多即可。
這裏將$m[i]$數組排序，以組數$ans$爲週期放入不同的$vector$數組中即可。

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const double eps = 1e-8;
const int NINF = 0xc0c0c0c0;
const int INF  = 0x3f3f3f3f;
const ll  mod  = 1e9 + 7;
const ll  maxn = 1e6 + 5;
const int N = 2e5 + 5;

int n,k,ans,c[N],m[N],cnt[N];
vector<int>G[N];

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>m[i];
		cnt[m[i]]++;
	}
	for(int i=1;i<=k;i++) cin>>c[i];
	for(int i=k;i>=1;i--) cnt[i]+=cnt[i+1],ans=max(ans,(cnt[i]+c[i]-1)/c[i]);
	sort(m+1,m+1+n);
	for(int i=1;i<=n;i++) G[i%ans].emplace_back(m[i]);
	cout<<ans<<'\n';
	for(int i=0;i<ans;i++){
		cout<<G[i].size();
		for(auto t:G[i]) cout<<" "<<t;
		cout<<'\n';
	}
	return 0;
}