E1. Reading Books (easy version)--------------------思维(贪心+模拟)

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
解析:
我们将书分为三种
第一种:只有Alice 喜欢
第二种:只有Bob 喜欢
第三种:Alice和Bob都喜欢

我们只需贪心地单独取就是第一种和第二种分别取。如果数量不够k个,那么我们就要从第三种里面开始取。
等取够k个的时候,我们再从第三种里面去找最优的情况,来代替单独取,使得答案最小

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+1000;
typedef long long ll;
ll a[N],b[N],c[N];
int n,k,t;
int cnt0,cnt1,cnt2;


int main()
{
	ll ans=0;
	cin>>n>>k;
	for(int i=1,f,f1;i<=n;i++)
	{
		cin>>t>>f>>f1;
		if(f==1&&f1==1)  c[++cnt2]=t;
		else if(f==1) a[++cnt0]=t;
		else if(f1==1) b[++cnt1]=t;
	}
	sort(a+1,a+1+cnt0);
	sort(b+1,b+1+cnt1);
	sort(c+1,c+1+cnt2);
	for(int i=1;i<=k;i++)  ans+=(a[i]+b[i]);//单独取
	cnt0=min(cnt0,k);cnt1=min(cnt1,k);
	int l=cnt0,r=cnt1;
	int m=min(l,r);
	int s=1;
	if(m<k)  //不够k个,从第三种里面取
	{
		if(k-m>cnt2) 
		{
			cout<<-1<<endl;
			return 0;
		}
		s=k-m+1;
		for(int i=1;i<=k-m;i++)
		{
			if(l<k&&r<k) l++,r++,ans+=c[i];
			else if(l<k)
			{
				l++;
				if(cnt1) ans=ans-b[cnt1--]+c[i];
			}
			else if(r<k)
			{
				r++;
				if(cnt0) ans=ans-a[cnt0--]+c[i];
			}
		}
	}
	for(int i=s;i<=cnt2;i++) //取完k个,从第三种挑选最优方案来代替单独取
	{
		ll sum=a[cnt0]+b[cnt1];
		if(sum>c[i])
		{
			ans=ans-sum+c[i];
			if(cnt1) cnt1--;
			if(cnt0) cnt0--;
		}
	}
	cout<<ans<<endl;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章