51Nod - 1050 循环数组最大子段和(dp)

描述

N个整数组成的循环序列a[1],a[2],a[3],…,a[n]，求该序列如a[i]+a[i+1]+…+a[j]的连续的子段和的最大值（循环序列是指n个数围成一个圈，因此需要考虑a[n-1],a[n],a[1],a[2]这样的序列）。当所给的整数均为负数时和为0。
例如：-2,11,-4,13,-5,-2，和最大的子段为：11,-4,13。和为20。

Input

第1行：整数序列的长度N（2 <= N <= 50000)
第2 - N+1行：N个整数 (-10^9 <= S[i] <= 10^9)

Output

输出循环数组的最大子段和。

Input示例

6
-2
11
-4
13
-5
-2

Output示例

20

思路

本题与普通的最大子段和问题不同的是，最大子段和可以是首尾相接的情况，即可以循环。那么这个题目的最大子段和有两种情况:

1. 正常数组中间的某一段和最大。这个可以通过普通的最大子段和问题求出
2. 此数组首尾相接的某一段和最大。这种情况是由于数组中间某段和为负值，且绝对值很大导致的，那么我们只需要把中间的和为负值且绝对值最大的这一段序列求出，用总的和减去它就行了

所以先对原数组求最大子段和，得到ans1，然后把数组中所有元素符号取反，再求最大子段和，得到ans2，原数组的所有元素和为ans，那么最终答案就是max(ans1, ans + ans2)

代码

#include <cstdio>
#include <cstring>
#include <cctype>
#include <stdlib.h>
#include <string>
#include <map>
#include <iostream>
#include <stack>
#include <cmath>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
const ll N=50000+10;
ll a[N],n;
ll maxnum(ll a[])
{
    ll maxx=0,ans=0;
    for(ll i=1; i<=n; i++)
    {
        ans+=a[i];
        if(ans<0)ans=0;
        maxx=max(maxx,ans);
    }
    return maxx;
}
int main()
{
    ll sum=0;
    scanf("%lld",&n);
    for(ll i=1; i<=n; i++)
    {
        scanf("%lld",&a[i]);
        sum+=a[i];
    }
    ll ans1=maxnum(a);
    for(ll i=1; i<=n; i++)
        a[i]=-a[i];
    ll ans2=maxnum(a);
    printf("%lld\n",max(ans1,sum+ans2));
    return 0;
}