原题链接:https://codeforces.com/contest/1077/problem/D
题目大意
给定n个数,从中选出k个数作为一个子集(允许有重复的数,如[7,1,3,1]),使得从n个数中可以“cut”出最多个上述的子集。
如给定n个数为[1,2,3,2,4,3,1],k=3,则子集可以选为[2,1,3],这样从n个数中“cut"出一个子集后,剩下:[2,4,3,1],再“cut"出一个子集后剩下[4],共可以“cut”出两个子集。
解法
可以二分猜答案check也可以贪心
二分答案
#include <iostream>
#include <ctime>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#define forn(i, n) for (int i = 0; i < (int)(n); ++i)
#define for1(i, n) for (int i = 1; i <= (int)(n); ++i)
#define pb push_back
using namespace std;
typedef long long LL;
const int N=200000+5;
int n,k,s[N],Hash[N];
bool check(int x){
int cnt=0;
for(int i=1;i<=200000;i++){
cnt+=Hash[i]/x;
}
return cnt>=k;
}
int main() {
// 2 4 2 1 1
// 4 2 2 1 1
// 2 1 1
#ifdef LOCAL_DEFINE
freopen("input.in", "rt", stdin);
#endif
scanf("%d%d",&n,&k);
for1(i,n){
scanf("%d",&s[i]);
Hash[s[i]]++;
}
int l=1,r=n;
while(r-l>5){
int mid=(l+r)/2;
// cout<<l<<" "<<mid<<" "<<r<<endl;
if(check(mid)){
l=mid;
}
else r=mid;
}
int ans=-1;
for(int i=r;i>=l;i--){
if(check(i)){
ans=i;
break;
}
}
int cnt=0;
for(int i=1;i<=200000;i++){
for(int j=1;j<=Hash[i]/ans;j++){
cout<<i<<" ";
cnt++;
if(cnt==k){
return 0;
}
}
}
cout<<endl;
#ifdef LOCAL_DEFINE
cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
return 0;
}
贪心与优先队列
挖坑待填……