【CF】1379-C-Choosing flowers 题解

传送门：1379-C
标签：思维

题目大意

给定m种花，第i种花有两个属性a_i和b_i，买一束该种花会获得a_i的价值，买两束即以上后除第一束花获得a_i价值外，其它每束花获得b_i的价值，你最多只能买n束花，问你能获得的最大价值是多少？
输入：两个正整数n(n<=1e9)和m(m<=1e6)，接下来m行每行两个正整数a_i和b_i。
输出：你能获得的最大价值。

算法分析

• 看到这种题目，我们肯定最先想到背包问题，但这题的每个物品占的价值是固定的，所以更简单，我们直接考虑贪心。假设有一种花的a_i和b_i都是最大的，那么我们肯定买n束这种花。如果有的花a_i大，有的花b_i大，那么我们显然只买一束a_i大的花，然后买很多束b_i大的花。这时候我们猜一个结论：最优策略中最多只有一种花的数量是大于等于二的。
• 这个结论也很好证明：假设有两种花（设为i和j）的数量都为x（x>=2），那么它们的贡献之和为a_i+a_j+xb_i+xb_j。这种情况下我们会发现b_i和b_j是可以相互替换的，我们将2x-2束花全部换成较大的那个显然是更优的，且不会影响其余花的价值。得到了这个结论我们尝试枚举每种花作为数量最多的那个，然后对每种情况取最大价值即可。
• 现在只要思考如何在枚举的过程中快速得到最大价值。既然其它种类的花最多取一束，且能获得的贡献固定为a_i，那我们只要从大到小取就行了。那取到什么时候为止呢？我们枚举到的花是数量最多的（假设为第k种花），那么就要保证其最少取一束，也就是说a_k是必取的，取其它种类花的数量只会改变b_k的数量，也就是说我们只要保证取的其它花的a_i大于b_k即可，这一点可以通过前缀和以及二分算法在log级别时间内实现。
• 如果每次枚举都对剩下m-1种花的a_i按降序排序，时间复杂度就会变成O(m²logm)，这是显然不行的。我们可以提前对所有花进行一次排序，然后枚举时考虑二分终点是否超过了k。这里分两种情况讨论：先假设二分结果取前x位。1、如果a_k小于或等于b_k，那么我们只要取前min(n-1,x)位。2、如果a_k大于b_k，那么二分结果肯定会把k包含进去，我们就相当于多算了一个a_k，这时候我们需要把这个多算的位置腾出来，如果此时x位还没取完（也就是n-1比x小），我们就接着取a_n，如果x位已经取完，我们就取b_k。这样一来总复杂度就稳定在O(mlogm)了。

代码实现

#include <iostream>
using namespace std;
#include <algorithm>
long long sum[100005];
struct Item{
	int x,y;
}I[100005];

bool rule(Item a,Item b){
	return a.x>b.x;
}

int main(){
	long long l,r,i,j,n,m,T,l1,r1,mid;
	long long mx,ans;
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	cin>>T;
	while(T--){
		cin>>n>>m;
		mx=0;
		for(i=1;i<=m;i++)
			cin>>I[i].x>>I[i].y;
		sort(I+1,I+m+1,rule);
		for(i=1;i<=m;i++)sum[i]=sum[i-1]+I[i].x;
		I[m+1].x=2e9;
		for(i=1;i<=m;i++){
			l=1;r=m+1;
			while(l<r){
				mid=(l+r)>>1;
				if(I[mid].x>I[i].y)l=mid+1;
				else r=mid;
			}
			ans=sum[min(n-1,l-1)]+I[i].x+1LL*I[i].y*max(n-l,0LL);
			if(i<=min(n-1,l-1)){
				ans-=I[i].x;
				if(n>=l)ans+=I[i].y;
				else ans+=I[n].x;
			}
			mx=max(mx,ans);
		}
		cout<<mx<<'\n';
	}
	return 0;
}