3627. 最大差值(Acwing)

给定一个长度为 n 的非负整数序列 a1,a2,…,an。

你可以对该序列进行最多 k 次操作。

每次操作选择两个非 0 的元素 ai 和  aj,然后选择一个整数 c(0≤c≤ai),使得 ai 减少 c,aj 增加 c。

请问,在操作全部完成后,序列中的最大值和最小值之差是多少。

例如,如果初始序列为 [5,5,5,5] 而 k=1,则一种最优方案是将 a2 减少 5,将 a4 增加 5,得到序列 [5,0,5,10],这样最大值和最小值之差为 10。

再例如,如果序列中的所有元素都为 0,则无法进行任何操作,所以最大值和最小值之差也为 0。

输入格式

第一行包含整数 T,表示共有 T 组测试数据。

每组数据第一行包含整数 n 和 k。

第二行包含 n 个整数 a1,a2,…,an。

输出格式

每组数据输出一行,一个整数,表示可以得到的最大差值。

数据范围

对于前三个测试点,1≤n≤10。
对于全部测试点,1≤T≤1000,1≤k<n≤2×105,0≤ai≤109,每个输入的 T 组数据的 n 之和不超过 2×105。

输入样例:

2
4 1
5 5 5 5
3 2
0 0 0

输出样例:

10
0

 由题意可以知道可以进行k次操作,为了获得最大差值,每次都对最大的两个值进行操作最大的数加上第二大的数,第二大的数变为0,一直进行K次操作这样可以得到最大差值,分析一下可以知道我们只需要对n个数进行排序,前k个大的数的和就是我们所求的最大差值

#include<iostream>
#include<algorithm>
using namespace std;
const int N=2e5+10;
int a[N];
int main(){
	int T;
	cin>>T;
	while(T--){
		int n,k;
		cin>>n>>k;
		for(int i=1;i<=n;i++)cin>>a[i];
		sort(a+1,a+n+1);
			long long sum=0;
			for(int i=n;i>=n-k;i--){
				sum+=a[i];
			}
			cout<<sum<<endl;
	}
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值