nyoj745-蚂蚁的难题(二)

蚂蚁的难题(二)

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

难度： 3

描述

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

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

输入

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

输出

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

样例输入

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

样例输出

5
7

题目思路：序列形成一个环，求最大子段和，最直接的想法是O(n^2)枚举每个起点求一次最大子段和，但数据过大这样明显会超时！

利用最大子段和原理，我们可以制造一个长度为2*n-1的伪环，即在序列n之后再接上序列中（1~n-1）这n-1个数。

这样每个数向后移动n次后都会回到自己，即形成了一个环。

形成一个环后，要分情况处理，因为若序列所有数的和大于零，若此时再求最大子段和，便会重复累计，但此时若是求最小子段和便不会跨越序列中的n个数，只要拿所有数之和sum - 此时的最小子段和max便得出这个环的最大字段和。

根据序列所有数的和判断，我们只要在这个环中直接求出最大子段和或者剔除最小子段和即可！。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define MAX 50010
const long long INF=10000000000;
long long a[MAX];
int main()
{
    int n;
    while (scanf("%d",&n)!=EOF)
    {
        int i,j;
        long long sum,max=0,ans=0;
        for (i=0;i<n;i++)
        {
            scanf("%lld",&a[i]);
            ans+=a[i];
        }
        if (ans<=0)
        {
            max=-INF;
            sum=0;
            for (i=0,j=0;i<2*n-1;i++,j++)
            {
                if (j>=n)
                    j=0;
                sum+=a[j];
                if (0>sum)
                {
                    sum=0;
                }
                if (max<sum)
                    max=sum;
            }
            printf("%lld\n",max);
        }
        else
        {
            sum=0;
            max=INF;
            for (i=0,j=0;i<2*n-1;i++,j++)
            {
                if (j>=n)
                    j=0;
                sum+=a[j];
                if (0<sum)
                {
                    sum=0;
                }
                if (max>sum)
                    max=sum;
            }
            printf("%lld\n",ans-max);
        }
    }
    return 0;
}