Codeforces Round #653 (Div. 3) E2 - Reading Books (hard version) (2500)

最新推荐文章于 2025-02-10 17:11:28 发布

狐佑怜

最新推荐文章于 2025-02-10 17:11:28 发布

阅读量280

点赞数

CC 4.0 BY-SA版权

分类专栏：题解暴力文章标签：算法

本文链接：https://blog.youkuaiyun.com/qq_43586463/article/details/107096184

题解同时被 2 个专栏收录

59 篇文章

订阅专栏

暴力

33 篇文章

订阅专栏

本文深入解析了一种基于暴力模拟的算法实现，通过选择特定数量的书籍以满足特定条件下的最优解，介绍了如何处理复杂的选择与交换策略，以达到最小化总时间的目标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Reading Books (hard version)
题意：
有 n 本书，每本书有3个属性，int类型的 time，bool类型的a,bool类型的b,现在你需要选择 m 本书，并且满足 a 为 1 的书至少要有 k 本，b 为 1 的书至少要有 k 本，在这个前提下 val 的总和还需要最小

思路：
暴力模拟，先全选 a==1&&b==1的书，然后再按照 5 种方案加书

用一本a==1&&b==0的书和一本a==0&&b==1的书与一本a==1&&b==1的书交换
加一本a==1&&b==0的书
加一本a==0&&b==1的书
加一本a==1&&b==1的书
加一本a==0&&b==0的书

代码附：

#pragma GCC optimize("Ofast","inline","-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 2e5+10;
const int inf=1e9;
struct node
{
    int num,tim;
} A[N],B[N],C[N],D[N];
bool cmp(node l1,node l2)
{
    return l1.tim<l2.tim;
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,m,k,x,y,z,a=0,b=0,c=0,d=0;
    cin>>n>>m>>k;
    for(int i=1; i<=n; ++i)
    {
        cin>>x>>y>>z;
        if(y&&z)
        {
            C[c].num=i;
            C[c++].tim=x;
        }
        else if(y)
        {
            A[a].num=i;
            A[a++].tim=x;
        }
        else if(z)
        {
            B[b].num=i;
            B[b++].tim=x;
        }
        else
        {
            D[d].num=i;
            D[d++].tim=x;
        }
    }
    if(a+c<k||b+c<k)//not enough
    {
        cout<<-1<<endl;
        return 0;
    }
    if(c<k&&(k-c)*2+c>m)//More than the least
    {
        cout<<-1<<endl;
        return 0;
    }
    sort(A,A+a,cmp);
    sort(B,B+b,cmp);
    sort(C,C+c,cmp);
    sort(D,D+d,cmp);
    int ai=0,bi=0,ci=0,di=0,sum=0;
    for(; ci<c; ++ci)//all both
    {
        if(ci==k)
            break;
        sum+=C[ci].tim;
    }
    while(ai+ci<k)
    {
        sum+=A[ai].tim+B[bi].tim;
        ai++,bi++;
    }
    while(ai+bi+ci+di<m)
    {
        int pro1,pro2,pro3,pro4,pro5;
        if(ai<a&&bi<b&&ci)
            pro1=A[ai].tim+B[bi].tim-C[ci-1].tim;
        else
            pro1=inf;
        if(ai<a)
            pro2=A[ai].tim;
        else
            pro2=inf;
        if(bi<b)
            pro3=B[bi].tim;
        else
            pro3=inf;
        if(ci<c)
            pro4=C[ci].tim;
        else
            pro4=inf;
        if(di<d)
            pro5=D[di].tim;
        else
            pro5=inf;
        int mini=min(pro1,min(pro2,min(pro3,min(pro4,pro5))));
        if(mini==inf)
        {
            cout<<-1<<endl;
            return 0;
        }
        sum+=mini;
        if(mini==pro1&&pro1<inf)
            ai++,bi++,ci--;
        else if(mini==pro2&&pro2<inf)
            ai++;
        else if(mini==pro3&&pro3<inf)
            bi++;
        else if(mini==pro4&&pro4<inf)
            ci++;
        else if(mini==pro5&&pro5<inf)
            di++;
    }
    cout<<sum<<endl;
    for(int i=0; i<ai; ++i)
        cout<<A[i].num<<" ";
    for(int i=0; i<bi; ++i)
        cout<<B[i].num<<" ";
    for(int i=0; i<ci; ++i)
        cout<<C[i].num<<" ";
    for(int i=0; i<di; ++i)
        cout<<D[i].num<<" ";
    return 0;
}