CDOJ1357--柱爷与最大区间和

传送门: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 
#include 
#include 
#include 
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;
}


你可能感兴趣的:(动态规划)