一开始读完这道题,我心想不就是找最小嘛,贪心试试!
结果就是连题目样例都没过……
题目的标签提示用前缀和,虽然这题求一个区间的元素和,感觉很像要用前缀和的样子,但是这题用前缀和该怎么写?之后我思来想去,但从未成功,最后妥协的去看了答案……
既然很难保证究竟是先删去前面两个最小的数好,还是先删去最后面最大的数好,那就挨个把所有的可能性遍历一遍!
一共就只有k次删除操作,要把k次操作都做完。(但我觉得题目这里有问题,他说在k次操作内保证和最大,那一次都不操作不肯定是最大?)那要是删除前面两个最小的数1次,就一定会删除后面最大的数k - 1次。反映在数组的下标中,就是n - k与自己定义的p位置的关系。k每次减1,p每次加2,反复循环寻找最大值,直到所有情况遍历结束。
总结:以后看到求一段数组区间的和的问题,不管一开始有没有思路,都要想到可能会用到前缀和。虽然可能一开始不知道怎么用,但是在没有具体思路,或者有思路但是不可行时,一定要试试用前缀和解题!
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int maxn = 2e5 + 9;
ll a[maxn];
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);
for (int i = 1; i <= n; i++) a[i] += a[i - 1];
int p = 0;
ll ans = 0;
while (k >= 0){
ans = max(ans, a[n - k] - a[p]);
p += 2;
k--;
}
cout << ans << endl;
}
return 0;
}