【Codeforces】CF1077D-Cutting Out(二分答案或贪心+优先队列)

原题链接: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 
#include 
#include 
#include 
#include 
#include 
#include 

#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<
		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;
}

贪心与优先队列

挖坑待填……

你可能感兴趣的:(算法与数据结构)