CDOJ-2016-B-柱爷与最大区间和

柱爷爱思考,凡事喜欢举一反三,常常能想到别人没想过的问题。
比如最大区间和这个问题:在一数列上选出一段区间,使得这段区间和最大。
柱爷想:如果选出两段区间(不相邻)会怎样呢?
柱爷很快想到了答案,你呢?

Input
第一行输入一个数N,表示数组的长度。
第二行输入N个数,表示各元素的值。

数据保证:
3≤N≤500000
−100

#include 
#define MAX(a, b) a > b ? a : b
#define MAXSIZE 500005
#define INF 0x3f3f3f3f

int main(int argc, const char * argv[])
{
    int i;
    int N;
    int A[MAXSIZE];
    int sumA, sumB, sumBB; //正序/逆序
    int ans;
    int B[MAXSIZE] = {0};

    while (~scanf("%d", &N))
    {
        ans = sumA = sumB = sumBB = -INF;

        for (i = 1; i <= N; i++)
        {
            scanf("%d", A + i);
        }

        B[N + 1] = -INF;
        for (i = N; i > 2; i--)
        {
            sumA = sumA > 0 ? sumA + A[i] : A[i];
            B[i] = MAX(B[i + 1], sumA);
        }
        for (i = 1; i < N - 1; i++)
        {
            sumB = sumB > 0 ? sumB + A[i] : A[i];
            sumBB = sumB > sumBB ? sumB : sumBB;

            if (sumBB + B[i + 2] > ans)
            {
                ans = sumBB + B[i + 2]; //中间隔着i + 1
            }
        }

        printf("%d\n", ans);
    }

    return 0;
}

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