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 <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;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章