CF 1374 E1. Reading Books (easy version)

传送门

题意：给定n本书，两个孩子A和B都至少需要读k本书，然后给定n本书的时间和两个孩子对这本书是否喜欢，问两个孩子都读了至少k本书的前提下最少的时间花费是多少？（如果这本书被选择，不论几个人读都是花费t的时间，不喜欢则说明不读）

思路：一开始想着像dp，但看了样例二发现是个简单的贪心，我们可以把书分成三类：AB喜欢，A喜欢，B喜欢，我们不妨把A喜欢和B喜欢贪心组合（即A+B的时间最少的组合）变成AB喜欢，我们最后只在AB喜欢中选，其他剩余的没意义，最后AB喜欢排个序，取前k小的AB喜欢即可。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <string>
#include <vector>
#include <cmath>
 
using namespace std;
 
#define ll long long
#define pb push_back
#define fi first
#define se second

const int N = 2e5 + 10;

void solve()
{      
    int n, k;
    cin >> n >> k;
    vector<int > a, b, ab;
    for(int i = 0; i < n; ++i){
        int t, x, y;
        cin >> t >> x >> y;
        if(x + y == 0) continue;
        if(x + y == 2) ab.pb(t);
        else if(x == 1) a.pb(t);
        else if(y == 1) b.pb(t);
    }

    sort(a.begin(), a.end());
    sort(b.begin(), b.end());
    int al = a.size();
    int bl = b.size();
    int ai, bi;
    ai = bi = 0;
    while(ai < al && bi < bl){
        ab.pb(a[ai++] + b[bi++]);
    }
    sort(ab.begin(), ab.end());
    int abl = ab.size();

    if(abl < k) cout << "-1" << endl;
    else{
        int Min = 0;
        for(int i = 0; i < k; ++i){
            Min += ab[i];
            //cout << "ab[i] = " << ab[i] << endl;
        }
        cout << Min << endl;
    }
}
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0); 
    solve();
 
    return 0;
}