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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值