傳送門: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;
}