传送门:http://acm.uestc.edu.cn/#/problem/show/1357
分析:首先,我们需要知道怎么求一段最大区间和。从左往右扫的时候,用pre记录当前的最大值,f[i]表示从最左端到i这个区间上的最大和。当扫到i时,如果pre已经小于0,那么直接将a[i]赋值给pre,否则加上a[i],然后f[i]=max(f[i-1], pre)。时间复杂度为O(n)。求两段的话,再倒着在扫一遍就行,接着枚举分界点,最终答案为ans = max(anx, f[k-1]+g[k+1])。
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int INF = 1<<29;
const int maxn = 500000+5;
int f[maxn], g[maxn];
int a[maxn];
int main() {
int n;
while(~scanf("%d", &n)) {
int pre = -INF;
f[0] = -INF;
for(int i = 0; i < n; i++) {
scanf("%d", &a[i]);
if(i == 0) f[i] = a[0], pre = a[0];
else {
pre = pre > 0 ? pre+a[i] : a[i];
f[i] = max(pre, f[i-1]);
}
}
int last = -INF;
g[n] = -INF;
for(int i = n-1; i >= 2; i--) {
last = last > 0 ? last+a[i] : a[i];
g[i] = max(last, g[i+1]);
}
int ans = -INF;
for(int k = 1; k < n-1; k++) {
ans = max(ans, f[k-1]+g[k+1]);
}
printf("%d\n", ans);
}
return 0;
}