题目描述

给定一个长度为 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次次大值，即为最大差值。
c取次大值， 让次大值变为0就相差最大了

时间复杂度 : O(n)

C++ 代码

#include<iostream>
#include<algorithm>

using namespace std;

const int N = 200010;

typedef long long LL;

int a[N];

int main()
{
    int t;
    cin >> t;

    while(t --)
    {
        int n, k;
        cin >> n >> k;
        for(int i = 0; i < n; i ++) cin >> a[i];

        sort(a, a + n);

        if(a[0]== a[n - 1] && a[0] == 0) cout << "0" << endl;
        else
        {
            LL res = a[n - 1];
            for(int i = n - 2; i >= 0 && k > 0; i --)
            {
                    res += max(a[i], 0);
                    k -= 1;

            }
            cout << res << endl;        
        }

    }
    return 0;
}

```