题目
给定一堆书,有只有Alice喜欢的,有只有Bob喜欢的,也有两者都喜欢的,选出Alice喜欢的k本书,Bob喜欢的k本书,两种选择可以有重复,组成一个新集合,要求新集合对应的总阅读时间最少,输出这个最少时间。
(如果这本书被选择,不论几个人读都是花费t的时间,不喜欢则说明不读)
基本思路
按阅读时间,自小到大,将只有Alice喜欢的排序,将只有Bob喜欢的排序,将两者自小到大配对,配成两者都喜欢的,即两本书的阅读时间,合并成一本书的阅读时间,归类为两者都喜欢的书,最后只在AB喜欢中选。
证明:只有Alice喜欢的书,与只有Bob喜欢的数量是一致的。
都喜欢的书有c本,只有Alice喜欢的有a本,只有Bob喜欢的有b本。
因Alice喜欢的书有k本,k=c+a;
因Bob喜欢的书有k本,k=c+b;
c+a=c+b
a=b.
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=2e5;
int a[maxn],b[maxn],c[maxn];
int main(){
ios::sync_with_stdio(false);
cin.tie(NULL);
int n,k,t,x,y;
int an,bn,cn;
an=bn=cn=0;
cin>>n>>k;
for(int i=0;i<n;i++){
cin>>t>>x>>y;
if(x&&y) c[cn++]=t;
else if(x) a[an++]=t;
else if(y) b[bn++]=t;
}
sort(a,a+an);
sort(b,b+bn);
int m=min(an,bn);
for(int i=0;i<m;i++)
c[cn++]=a[i]+b[i];
sort(c,c+cn);
if(cn<k){cout<<-1<<endl;return 0;}
int ans=0;
for(int i=0;i<k;i++) ans+=c[i];
cout<<ans<<endl;
return 0;
}