题意:
给你一个长为n的正整数数组,你要选一个连续子序列,对其每个元素加上正号或者负号,使其和为0,求方案数
题解:
dp( i , j , k )表示dp到第i个,和为j,这个元素是选了正还是负
滚动数组优化
1 #include<bits/stdc++.h> 2 #define ll long long 3 #define maxn 1005 4 #define maxm 10005 5 using namespace std; 6 const ll mod = 1000000007; 7 int n,a[maxn]; 8 ll dp[2][3][maxm*2]; 9 int main() 10 { 11 scanf("%d",&n); 12 for(int i=1;i<=n;++i)scanf("%d",&a[i]); 13 ll ans=0; 14 for(int i=1;i<=n;++i) 15 { 16 int now=i&1,pre=now^1; 17 memset(dp[now],0,sizeof(dp[now])); 18 dp[now][1][10005+a[i]]=1;dp[now][2][10005-a[i]]=1; 19 for(int k=1;k<=2;++k) 20 for(int j=0;j<maxm*2;++j)if(dp[pre][k][j]) 21 { 22 dp[now][1][j+a[i]]=(dp[now][1][j+a[i]]+dp[pre][k][j])%mod; 23 dp[now][2][j-a[i]]=(dp[now][2][j-a[i]]+dp[pre][k][j])%mod; 24 } 25 for(int k=1;k<=2;++k)ans=(ans+dp[i&1][k][10005])%mod; 26 } 27 printf("%I64d ",ans); 28 return 0; 29 }