1. 题目来源
链接:3627. 最大差值
2. 题目解析
贪心题,有点考思维。
每次操作肯定是将操作数直接减为 0,然后将这个操作数的值累加到最大值上面。
那么贪心来讲,我们要让最大值尽可能大,让每次累加的数也尽可能大,那肯定是累加除过最大值的前 k
大的数。
故问题就转化为求序列前 k+1
大数的和。
本题需要开 long long
。
时间复杂度: O ( n l o g n ) O(nlogn) O(nlogn)
空间复杂度: O ( n ) O(n) O(n)
正常写法:
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 2e5 + 5;
int n, k;
int a[N];
int main() {
int T;
scanf("%d", &T);
while (T -- ) {
scanf("%d%d", &n, &k);
for (int i = 0; i < n; i ++ ) scanf("%d", &a[i]);
sort(a, a + n, greater<int>());
LL res = 0;
for (int i = 0; i < k + 1; i ++ ) res += a[i];
printf("%lld\n", res);
}
return 0;
}