当前位置: 代码迷 >> 综合 >> Codeforces E1. Reading Books (easy version) (模拟) (Round #653 Div.3)
  详细解决方案

Codeforces E1. Reading Books (easy version) (模拟) (Round #653 Div.3)

热度:40   发布时间:2023-12-22 13:44:41.0

传送门

题意: 现有n本书,每本书有三个性质:ai 阅读所需时间,bi = 1 表示Alice 喜欢该书, ci = 1 表示Bob喜欢该书。试问能否找到最好方案—Alice和Bob喜欢的数都至少有k本,且总阅读时间最短;若没有可行方案输出-1.
在这里插入图片描述
思路:

  • 先将所有书分为三个组:both两个人都喜欢的书,alice只有Alice喜欢的书,bob只有Bob喜欢的书;并将三个组按升序排序。
  • 如果不能满足k本书的条件直接输出-1
  • 再取最小的方案:先考虑both的数量不够k本或both中的时间比alice[i] + bob[i]大的情况,就取alice[i] + bob[i];反之就是直接取both[i]啦。

代码实现:

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cctype>
#include <cstring>
#include <iostream>
#include <sstream>
#include <string>
#include <list>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <functional>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {
    {
    1, 0}, {
    -1, 0}, {
    0, 1}, {
    0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double Pi = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;int n, k, ans;
int a, b, c;
int h1, h2, t1, t2, t3;
int both[N], alice[N], bob[N];signed main()
{
    IOS;cin >> n >> k;for(int i = 0; i < n; i ++){
    cin >> a >> b >> c;if(b && c)both[t1 ++] = a;else if(b && !c)alice[t2 ++] = a;else if(!b && c)bob[t3 ++] = a;}sort(both, both + t1);sort(alice, alice + t2);sort(bob, bob + t3);if(t1 + t2 < k || t1 + t3 < k){
    //cout << t2 << " " << t3 << endl;cout << -1 << endl;return 0;}while(h1 + h2 < k){
    if(h1 >= t1 || (h2 < min(t2, t3) && both[h1] > alice[h2] + bob[h2])){
    ans += alice[h2] + bob[h2];h2 ++;}else{
    ans += both[h1];h1 ++;}}cout << ans << endl;return 0;
}
  相关解决方案