Codeforces 383D. Antimatter DP



D. Antimatter
time limit per test
1 second
memory limit per test
256 megabytes
input
standard input
output
standard output

Iahub accidentally discovered a secret lab. He found there n devices ordered in a line, numbered from 1 to n from left to right. Each devicei (1 ≤ i ≤ n) can create either ai units of matter or ai units of antimatter.

Iahub wants to choose some contiguous subarray of devices in the lab, specify the production mode for each of them (produce matter or antimatter) and finally take a photo of it. However he will be successful only if the amounts of matter and antimatter produced in the selected subarray will be the same (otherwise there would be overflowing matter or antimatter in the photo).

You are requested to compute the number of different ways Iahub can successful take a photo. A photo is different than another if it represents another subarray, or if at least one device of the subarray is set to produce matter in one of the photos and antimatter in the other one.

Input

The first line contains an integer n (1 ≤ n ≤ 1000). The second line contains n integers a1a2, ..., an (1 ≤ ai ≤ 1000).

The sum a1 + a2 + ... + an will be less than or equal to 10000.

Output

Output a single integer, the number of ways Iahub can take a photo, modulo 1000000007 (109 + 7).

Sample test(s)
input
4
1 1 1 1
output
12
Note

The possible photos are [1+, 2-], [1-, 2+], [2+, 3-], [2-, 3+], [3+, 4-], [3-, 4+], [1+, 2+, 3-, 4-], [1+, 2-, 3+, 4-], [1+, 2-, 3-, 4+], [1-, 2+, 3+, 4-], [1-, 2+, 3-, 4+] and [1-, 2-, 3+, 4+], where "i+" means that the i-th element produces matter, and "i-" means that the i-th element produces antimatter.




import java.util.*;

public class CF383D{
    
    final int MOD = 1000000007;
    final int BASE = 10010;
    int[] a = new int[1111];
    int n,sum=0;
    int[][] dp = new int[1111][21111];
    
    CF383D(){
        Scanner in = new Scanner(System.in);
        n=in.nextInt();
        for(int i=1;i<=n;i++){
            a[i]=in.nextInt();
            sum+=a[i];
        }
        
        dp[0][BASE]=1;
        
        for(int i=1;i<=n;i++)
        {
            dp[i][BASE]=1;
            for(int j=-sum;j<=sum;j++)
            {
                if(dp[i-1][j+BASE]!=0)
                {
                    dp[i][j+BASE+a[i]]+=dp[i-1][j+BASE];
                    dp[i][j+BASE+a[i]]%=MOD;
                    dp[i][j+BASE-a[i]]+=dp[i-1][j+BASE];
                    dp[i][j+BASE-a[i]]%=MOD;
                }
            }
        }

        int ans=0;
        for(int i=1;i<=n;i++)
        {
            ans+=dp[i][BASE]-1;
            ans%=MOD;
        }
        
        System.out.println(ans);
    }
    
    public static void main(String[] args){
        new CF383D();
    }
}



你可能感兴趣的:(Codeforces 383D. Antimatter DP)