nyoj745_蚂蚁的难题(二)

蚂蚁的难题(二)

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 3

描述

下雨了，下雨了，蚂蚁搬家了。

已知有n种食材需要搬走，这些食材从1到n依次排成了一个圈。小蚂蚁对每种食材都有一个喜爱程度值Vi，当然，如果Vi小于0的时候，表示蚂蚁讨厌这种食材。因为马上就要下雨了，所以蚂蚁只能搬一次，但是能够搬走连续一段的食材。时间紧急，你快帮帮小蚂蚁吧，让它搬走的食材喜爱值和最大。

输入

有多组测试数据（以EOF结尾）。
每组数据有两行，第一行有一个n，表示有n种食材排成了一个圈。（1 <= n<= 50000)
第二行分别有n个数，代表蚂蚁对第n种食材的喜爱值Vi。(-10^9 <= Vi <= 10^9)

输出

输出小蚂蚁能够搬走的食材的喜爱值总和的最大。

样例输入

3
3 -1 2
5
-8 5 -1 3 -9

样例输出

5
7

一开始的思路是开一个2n的数组，把1-n的数据复制到n+1到2n，在求最大子段和，后来发现是错误的

max（直线型的最大子段和，所以的和-直线型的最小子段和）

#include <iostream>
#include <cstdio>
#define N 100010
#define INF 0x3f3f3f3f
using namespace std;

int a[N];

int main()
{
    int n;
    while (scanf("%d", &n) != EOF)
    {
        long long sum = 0;
        for (int i = 0; i < n; i++)
        {
            scanf("%d", &a[i]);
            sum += a[i];
        }
        long long sum1 = 0,sum2 = 0,  mmax = -INF, mmin = INF;
        for (int i = 0;  i < n; i++)
        {
            if(sum1 < 0)
                sum1 = a[i];
            else 
                sum1 += a[i];
            if (sum1 > mmax)
                mmax = sum1;
            if (sum2 > 0)
                sum2 = a[i];
            else 
                sum2 += a[i];
            if(sum2 < mmin)
                mmin = sum2; 

        }
        long long r = max(mmax, sum - mmin);
        printf("%lld\n", r);

    }

    return 0;
}