Codeforces Round #653 (Div. 3)-E1. Reading Books

题目链接
题意:
这里一共有n本书,A,B都分别喜欢其中的一部分,A和B读每本书都需要一些时间,如果这本书他们两个都喜欢,那么他们可以一起读,问你能否找到一部分书使得A喜欢k本,B也喜欢k本且读完用时最少。
思路:
开三个数组,分别存A喜欢的和B喜欢的和AB都喜欢的,然后按照时间大小排序,然后按照时间从小到大选取即可。
代码:

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=2e5+5;
const int mod=998244353;
const int inf=0x7fffffff;
const double pi=3.1415926535;
using namespace std;
int n,k,arr1[N],arr2[N],arr3[N],sum1=0,sum2=0,sum3=0;
int cmp(int a,int b)
{
    return a<b;
}
signed main()
{
    IOS;
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        if(b==1&&c==1)
        {
            arr1[sum1]=a;
            sum1++;
        }
        else if(b==1)
        {
            arr2[sum2]=a;
            sum2++;
        }
        else if(c==1)
        {
            arr3[sum3]=a;
            sum3++;
        }
    }
    if(sum1+min(sum2,sum3)<k)
    {
        cout<<-1<<endl;
    }
    else
    {
        sort(arr1,arr1+sum1,cmp);
        sort(arr2,arr2+sum2,cmp);
        sort(arr3,arr3+sum3,cmp);
        int num1=0,num2=0,num=0;
        while(num1+num2<k)
        {
            if(arr1[num1]>=arr2[num2]+arr3[num2]&&num2<min(sum2,sum3)||num1>=sum1)
            {
                num+=arr2[num2]+arr3[num2];
                num2++;
            }
            else
            {
                num+=arr1[num1];
                num1++;
            }
        }
        cout<<num<<endl;
    }
	return 0;
}
©️2020 CSDN 皮肤主题: 1024 设计师: 上身试试 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值