lanqiao 3260 - 最大数组和

代码及思路如下

#include <bits/stdc++.h>
using namespace std;

#define ll long long
const ll N = 2 * 1e5 + 10;

bool cmp(ll a, ll b) {
	return a < b;
}

ll arr[N] = { 0 };
ll prefix[N] = { 0 };//与搬石头题不同，这里的后缀和可用前缀和转化得到

//无法用贪心得到答案，因为你删除的数据会对后续的操作产生影响
/*
以样例
6 2
10 11 12 13 15 22为例
贪心的答案就是40
正解为46 删15 22
*/
int main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int t; cin >> t;
	while (t--) {
		ll n, k; cin >> n >> k;

		for (ll i = 1; i <= n; i++)
			cin >> arr[i];

		sort(arr, arr + n + 1, cmp);

		for (ll i = 1; i <= n; i++) {
			prefix[i] = prefix[i - 1] + arr[i]; 
		}

		//删除的顺序对结果没有影响，故可以遍历k+1次(存在前或后一次都不删的情况)
		//分别看删去a次前两个与k-a次后一个的答案哪个更大
		
		ll maxAns = prefix[n - k];
		for (ll i = 0; i <= k; i++) {
			ll temp;
			if (i * 2 <= n) {
				temp = prefix[n - (k - i)] - prefix[i * 2];
			}
			else
				temp = prefix[n - k];
			maxAns = max(maxAns, temp);
		}

		cout << maxAns << '\n';
	}
	return 0;
}

这种多次访问数组区间的题目考虑用前缀和解题