[CF799E]Aquarium decoration

Aquarium decoration

题解

很容易发现这是个贪心的题

由于直接贪没有什么思路，还是笔者太菜了，我们可以先将所有的材料分成四个部分：

AB都喜欢，A喜欢B不喜欢，B喜欢A不喜欢，AB都不喜欢。

然后，枚举第一部分的选择数量，求出对应的最小值，这部分可以贪心求出。

这样的时间复杂度是的，明显会T，但我们又发现最小值是可以线性推出的。

先将第1部分排序，那在第一部分选的i个一定是选的前i个。于是，在满足两人都喜欢的达到k个的情况下，即第2，3部分都选了个的情况下，就在第2，3，4部分下将剩下还能选的名额分配给现在最小的没选的材料。

每次i增加后，就在第2，3部分根据情况减少一部分，再将其凑满，如果有多的，就根据情况减少即可。

这样时间复杂度就是了，因为还要排序。

源码

好难调啊

 

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
using namespace std;
#define MAXN 500005
typedef long long LL;
#define int LL
typedef pair<LL,int> pii;
const int INF=0x7f7f7f7f7f7f7f7f;
int n,m,k,c[MAXN],la,lb,ans;
bool visa[MAXN],visb[MAXN];
vector<int> vec[5];
signed main(){
	scanf("%lld %lld %lld",&n,&m,&k);
	for(int i=1;i<=n;i++)scanf("%lld",&c[i]);
	scanf("%lld",&la);for(int i=1,x;i<=la;i++)scanf("%lld",&x),visa[x]=1;
	scanf("%lld",&lb);for(int i=1,x;i<=lb;i++)scanf("%lld",&x),visb[x]=1;
	for(int i=1;i<=n;i++){
		if(visa[i]&&visb[i]){vec[1].push_back(c[i]);continue;}
		if(visa[i]&&!visb[i]){vec[2].push_back(c[i]);continue;}
		if(!visa[i]&&visb[i]){vec[3].push_back(c[i]);continue;}
		if(!visa[i]&&!visb[i]){vec[4].push_back(c[i]);continue;}
	}
	for(int i=1;i<=4;i++)sort(vec[i].begin(),vec[i].end());
	int siz=vec[1].size(),sum=0,id1=0,id2=0,id3=0;ans=INF;
	if(la<k||lb<k){puts("-1");return 0;}
	if(2*k-siz>m||k>m){puts("-1");return 0;}
	for(int i=0;i<=siz;i++){
		if(i)sum+=vec[1][i-1];if(2*k-i>m)continue;if(i>m)break;
		if(vec[2].size()+i<k||vec[3].size()+i<k)continue;
		if(i+vec[2].size()+vec[3].size()+vec[4].size()<m)continue;
		while(i+id1<k)sum+=vec[2][id1],id1++;
		while(i+id2<k)sum+=vec[3][id2],id2++;
		//printf("%d %d %d\n",i,id1,id2);
		while(id1&&id1+i>k){
			int s1=vec[2][id1-1],s2=INF,s3=INF;
			if(id2<vec[3].size())s2=vec[3][id2];
			if(id3<vec[4].size())s3=vec[4][id3];
			if(s2>s1&&s3>s1)break;
			if(s2<=s3)sum-=s1-s2,id1--,id2++;
			else if(s3<s2)sum-=s1-s3,id1--,id3++;
		}
		while(id2&&id2+i>k){
			int s1=INF,s2=vec[3][id2-1],s3=INF;
			if(id1<vec[2].size())s1=vec[2][id1];
			if(id3<vec[4].size())s3=vec[4][id3];
			if(s1>s2&&s3>s2)break;
			if(s1<=s3)sum-=s2-s1,id1++,id2--;
			else if(s3<s1)sum-=s2-s3,id3++,id2--;
		}
		while(i+id1+id2+id3>m){
			int s1=0,s2=0,s3=0;
			if(id1&&i+id1>k)s1=vec[2][id1-1];
			if(id2&&i+id2>k)s2=vec[3][id2-1];
			if(id3)s3=vec[4][id3-1];
			if(s1>=s2&&s1>=s3)sum-=s1,id1--;
			else if(s2>=s1&&s2>=s3)sum-=s2,id2--;
			else if(s3>=s1&&s3>=s2)sum-=s3,id3--;
		}
		while(i+id1+id2+id3<m){
			int s1=INF,s2=INF,s3=INF;
			if(id1<vec[2].size())s1=vec[2][id1];
			if(id2<vec[3].size())s2=vec[3][id2];
			if(id3<vec[4].size())s3=vec[4][id3];
			//printf("%d %lld %lld %lld\n",i,s1,s2,s3);
			if(s1<=s2&&s1<=s3)sum+=s1,id1++;
			else if(s2<=s1&&s2<=s3)sum+=s2,id2++;
			else if(s3<=s1&&s3<=s2)sum+=s3,id3++;
		}
		//printf("%d %d %d %d:%d\n",i,id1,id2,id3,sum);
		ans=min(ans,sum);
	}
	printf("%lld\n",ans>INF-1?-1:ans);
	return 0;
}

谢谢！！！