给定一个长度为 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;
}