题目链接
题目大意
给你n本数，俩个人对这n本书有的喜欢，有的讨厌，现在问，这俩个人分别从他们喜欢的书中挑k本出来(如果挑不出来，输出-1)，现在问每本书的最少阅读时间的总和为多少?
题目思路
emmmm,我还以为要用优先队列啥的，没想到居然就是直接模拟一下就行了。
1.我们开三个数组分别存储:俩个小孩都喜欢的书，第一个小孩喜欢的书，第二个小孩喜欢的书的所需要看的时间。
2.再把时间从小到大排序，然后再分情况: (1)如果有一个人不能选k本书出来看，那么直接输出-1
(2)如果俩个人都喜欢的书的花费时间 大于 第一个小孩喜欢的书的时间加上第二个小孩喜欢的书的时间(书按照现在最短的那个时间依次推)，那么我们肯定是选择时间少的那种，然后把这俩本书删除，再判断。
(3)与第二种情况相反，那么我们就选择俩个人都喜欢的书，然后把这本书删除，即可
代码
#include<bits/stdc++.h>
using namespace std
;
typedef long long ll
;
const int maxn
=2e5+5;
const int mod
=1e9+7;
int n
,k
,a1
[maxn
],a2
[maxn
],a3
[maxn
],num1
,num2
,num3
;
int main(){
scanf("%d%d",&n
,&k
);
for(int i
=1,t
,a
,b
;i
<=n
;i
++){
scanf("%d%d%d",&t
,&a
,&b
);
if(a
&&b
){
a1
[++num1
]=t
;
}else if(a
&&(!b
)){
a2
[++num2
]=t
;
}else if((!a
)&&b
){
a3
[++num3
]=t
;
}
}
sort(a1
+1,a1
+1+num1
);
sort(a2
+1,a2
+1+num2
);
sort(a3
+1,a3
+1+num3
);
if(num1
+min(num2
,num3
)<k
){
printf("-1\n");
}else{
int pos1
=1,pos2
=1,sum
=0;
while(pos1
+pos2
-2<k
){
if(pos1
>num1
){
sum
+=a2
[pos2
]+a3
[pos2
];
pos2
++;
}else if(pos2
<=min(num2
,num3
)&&a1
[pos1
]>=a2
[pos2
]+a3
[pos2
]){
sum
+=a2
[pos2
]+a3
[pos2
];
pos2
++;
}else{
sum
+=a1
[pos1
];
pos1
++;
}
}
printf("%d\n",sum
);
}
return 0;
}
