Codeforces Round #653 (Div. 3) E1.Reading Books (easy version) (思维，sort)

题目传送
题意:
给你n本数，俩个人对这n本书有的喜欢，有的讨厌，现在问，这俩个人分别从他们喜欢的书中挑k本出来(如果挑不出来，输出-1)，现在问每本书的最少阅读时间的总和为多少?

思路:
看清楚了是每本书的最少阅读时间总和，害的我最开始没有看懂第二个样例。。。

1.我们开三个数组分别存储:俩个小孩都喜欢的书，第一个小孩喜欢的书，第二个小孩喜欢的书的所需要看的时间。

2.再把时间从小到大排序，然后再分情况:
(1)如果有一个人不能选k本书出来看，那么直接输出-1

(2)如果俩个人都喜欢的书的花费时间 大于 第一个小孩喜欢的书的时间加上第二个小孩喜欢的书的时间(书按照现在最短的那个时间依次推)，那么我们肯定是选择时间少的那种，然后把这俩本书删除，再判断。

(3)与第二种情况相反，那么我们就选择俩个人都喜欢的书，然后把这本书删除，即可

AC代码

#include <bits/stdc++.h>
inline long long read(){char c = getchar();long long x = 0,s = 1;
while(c < '0' || c > '9') {if(c == '-') s = -1;c = getchar();}
while(c >= '0' && c <= '9') {x = x*10 + c -'0';c = getchar();}
return x*s;}
using namespace std;
#define NewNode (TreeNode *)malloc(sizeof(TreeNode))
#define Mem(a,b) memset(a,b,sizeof(a))
#define lowbit(x) (x)&(-x)
const int N = 2e5 + 10;
const long long INFINF = 0x7f7f7f7f7f7f7f;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-7;
const int mod = 1e9+7;
const double II = acos(-1);
const double PP = (II*1.0)/(180.00);
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> piil;
ll arr1[N],arr2[N],arr3[N],num1,num2,num3,ans1,ans2,sum;
signed main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    ll n,k;
    cin >> n >> k;
    for(ll i = 1;i <= n;i++)
    {
        ll a,b,c;
        cin >> a >> b >> c;
        if(b && c)
            arr1[num1++] = a;
        else if(b && !c)
            arr2[num2++] = a;
        else if(!b && c)
            arr3[num3++] = a;//分别存储三种情况
    }
    sort(arr1,arr1+num1);sort(arr2,arr2+num2);sort(arr3,arr3+num3);//排序
    if(num1 + min(num2,num3) < k) cout << -1 << endl;//如果有人不能选k本书，那么直接输出-1
    else
    {
        while(ans1+ans2 < k)
        {
           if(ans1 >= num1)//如果俩个人都喜欢的书已经没有了，那么肯定只能分开选书了
           {
               sum = sum + arr2[ans2] +  arr3[ans2];
               ans2++;
           }
           else if(ans2 < min(num2,num3) && arr1[ans1] > arr2[ans2] + arr3[ans2])//前提是他们还有俩本书可以选
           {
               sum = sum + arr2[ans2] +  arr3[ans2];
               ans2++;
           }
           else//否则就是俩个人都喜欢的书的花费时间小于分开选书
           {
               sum += arr1[ans1];
               ans1++;
           }
        }
        cout << sum << endl;
    }
}