解析:
我們將書分爲三種
第一種:只有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;
}