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