CF1622C Set or Decrease

题目地址

https://codeforces.com/problemset/problem/1622/C

题目描述

略

解题思路

对于某个数$a_i$有两个操作：

操作1：$a_i$自减使得$a_i=a_i-1$

操作2：赋值$a_i=a_j$

求最少操作的次数，使得 $\sum _{i=1}^n{a}_i\le k$.

• 对于操作2，肯定是最小的赋值赋给较大值操作的次数会是最少
• 对于操作1，对最小的值自减，这样赋值给其它较大值时操作的次数也是最少的，更快地接近$k$.

利用上面的贪心思路，对数组$a$从小到大排序，对$a_1$做操作1，从$a_n$开始做操作2.

如何求出操作1和操作2的次数？

• 设数组区间和 $Sum(a,b)=\sum _{i=a}^b{a}_i$.
• 设操作1的次数为 $x$.
• 操作2的次数为 $y$ .

于是有了下面的表达式：
$$(a_1-x)\ast (y+1)+Sum(2,n-y)<=k$$

比较容易的方法当然是枚举所有的$x$和$y$，但根据题中给的数据范围，必然会超时。所以只能枚举其中的一个变量，根据数据范围，枚举$y$是比较明智的，把式子简单转化一下，有：

$(a_1-x)\ast (y+1)+Sum(2,n-y)<=k$
$(a_1-x)\ast (y+1)<=k-Sum(2,n-y)$
$a_1\ast y+a_1-x\ast y-x<=k-Sum(2,n-y)$

整理一下，最终得到:

$x>=\frac{-(k-Sum(2,n-y)-a_1\ast y-a_1)}{y+1}$

通过枚举$y$，直接求出$x$即可.

数据类型卡了好几次，反正无脑上long long就是了。

参考代码

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
const int MAXN=200005;
long long a[MAXN];
long long s[MAXN];
int main(){
	long long t,n,k,ans;
	cin>>t;
	while(t--){
		cin>>n>>k;
		ans= 0x7fffffff;
		for(int i=1;i<=n;i++)
			cin>>a[i];	
		
		sort(a+1,a+1+n);
		for(int i=1;i<=n;i++){
			s[i]=a[i];s[i]+=s[i-1];
		}
		
		if(s[n]<=k) {
			cout<<0<<endl;
			continue;
		}
		
		for(int y=0;y<n;y++)
		{
			long long temp=k-s[n-y]+s[1]-a[1]-a[1]*y;
			long long x=ceil(-1.0*(long long)(temp)/(y+1));
			//printf("x:%d,y:%d\n",x,y);
			//printf("res:%d\n",(a[1]-x)*(y+1)+s[n-y]-s[1]);
			
			//x为负数时表示不需要减任何数也能满足条件
			if(x<=0) x=0; 
			if(x+y<ans) ans=x+y;
		}
		cout<<ans<<endl;
		
	}
	return 0;
}