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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章