CODEFORCES, 383D Antimatter

题意：给出n个数，每个数>0的数组。选择一个连续的序列，给序列中每个数选择正负号，使得求和结果为0。问有多少种不同的选择。给出约束条件 a1+a2+...+an-1+an的值不超过10000
n<1000。

策略：如果暴力搜索枚举的话首先区间就能枚举1000+999+..+1  是10^6级别，再算上每个数有2种选择的话显然量级太大。

但是我们有一个关键的约束条件。在做的时候并没有注意，所以也是想了很久。

如果n个数的和不超过10000。那么也就意味着我们能够启发式的来进行搜索。也就是动态规划。

即我们如果已经知道c个数的和从[-10000~10000]，并且是以第c个数结尾的选法数，我们再取第c+1个数，我们就可以知道这一个数结尾可以组成哪些值，并且有多少种方法。

如果设dp[i][j]为前i个数且以第i个数结尾，和为j的种类的数。那么我们可以得到状态转移的方程：

dp[i][ j+a[i+1] ] +=dp[i][j]; 

dp[i][ j-a[i+1] ] +=dp[i][j];

其中i<1000,j<20000 

复杂度为10^7

最后的答案我们只要知道dp[i][0]的个数就可以了。

//Author:Wenjun Shi
//on 2016/7/14
#include <iostream>

#define MAX_N 1005
#define MOD 1e9+7
#define ZERO 10000
using namespace std;

int dp[MAX_N][20005];
int a[MAX_N];
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++)
    {
        dp[i][ZERO+a[i]]+=1;
        dp[i][ZERO-a[i]]+=1;
        for(int j=-10000;j<=10000;j++)
        {
            if(dp[i-1][ZERO+j]!=0){
              dp[i][ZERO+j+a[i]]+=(dp[i-1][ZERO+j]);
              if(dp[i][ZERO+j+a[i]]>MOD) dp[i][ZERO+j+a[i]]-=MOD;
              dp[i][ZERO+j-a[i]]+=(dp[i-1][ZERO+j]);
              if(dp[i][ZERO+j-a[i]]>MOD) dp[i][ZERO+j-a[i]]-=MOD;
            }
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        ans+=dp[i][ZERO];
        if(ans>MOD) ans-=MOD;
    }
    cout<<ans<<endl;
    return 0;
}