[ABC312F] Cans and Openers 题解

传送门

好题，但是建议放E题位置

题目大意

有 $n$ 个物品，现要从中选 $m$ 个，物品分为 $3$ 类。

第 $1$ 类：直接使用，获得 $a_i$ 喜悦值。

第 $2$ 类：需要用第 $3$ 类物品打开，打开后获得 $a_i$ 喜悦值。

第 $3$ 类：可一共打开 $a_i$ 个第 $2$ 类物品。

求最大能获得的喜悦值。

解题思路

赛时想了个假的贪心

我们可以枚举第 $2$ 类物品的数量。考虑选 $x$ 个第 $2$ 类物品时需要多少个第 $3$ 类物品。显然这个可以通过处理一个前缀和二分得到。当第 $2$ 类和第 $3$ 类物品个数确定时，可以贪心地取第 $1$ 类物品。于是问题就解决了。

代码示例

#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,m,a0[200010],a1[200010],a2[200010];
int cnt0=0,cnt1=0,cnt2=0,ans=0;
int num0[200010],num1[200010],num2[200010];
signed main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		int t,x;
		cin>>t>>x;
		if(t==0) a0[++cnt0]=x;
		else if(t==1) a1[++cnt1]=x;
		else a2[++cnt2]=x;
		//分三类 
	}
	sort(a0+1,a0+cnt0+1),sort(a1+1,a1+cnt1+1),sort(a2+1,a2+cnt2+1);
	
	for(int i=1;i<=cnt0/2;i++) swap(a0[i],a0[cnt0-i+1]);
	for(int i=1;i<=cnt1/2;i++) swap(a1[i],a1[cnt1-i+1]);
	for(int i=1;i<=cnt2/2;i++) swap(a2[i],a2[cnt2-i+1]);
	//倒转处理，先贪心地取最大的 
	
	for(int i=1;i<=cnt0;i++) num0[i]=num0[i-1]+a0[i]; 
	for(int i=1;i<=cnt1;i++) num1[i]=num1[i-1]+a1[i];
	for(int i=1;i<=cnt2;i++) num2[i]=num2[i-1]+a2[i];
	//预处理前缀和 
	
	for(int i=0;i<=cnt1;i++){
		if(i>num2[cnt2]) continue;//第3类物品不够
		int it=lower_bound(num2+1,num2+cnt2+1,i)-num2;
		if(i==0) it=0;
		if(i+it>m) continue;//超过m个
		int Max=min(m-i-it,cnt0);//没写这个多挂了一发。。。 
		ans=max(ans,num1[i]+num0[Max]);//取i个2类，Max个1类 
	}
	cout<<ans<<endl;
	return 0;
}