hdu5800_dp

http://acm.hdu.edu.cn/showproblem.php?pid=5800

题意：

给定一个由n个元素组成的序列,和s (n<=1000,s<=1000)

　　求 ： 

题解的做法是f[i][j][s1][s2]表示前i个数总和为j必选s1个必不选s2个的方案数，这样是O(n*s*4)的。

　　对于每一个数，有4种选法：选，不选，必选，必不选，然后转移就好了。

　　答案就是sigma(f[n][i][2][2]) ,i∈[0,s]。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <ctime>
#include <queue>
#include <list>
#include <set>
#include <map>
using namespace std;
#define mod 1000000007
typedef long long LL;

int dp[1010][1010][4][4];
int a[1010];
void Add(int &x, int y)
{
    x = (x + y) % mod;
}
int main()
{
    int n, s, t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d %d", &n, &s);
        for(int i = 1; i <= n; i++)
            scanf("%d", a + i);
        memset(dp, 0, sizeof(dp));
        dp[0][0][0][0] = 1;
        for(int i = 1; i <= n; i++)
        {
            for(int j = 0; j <= s; j++)
            {
                for(int k = 0; k <= 2; k++)
                {
                    for(int l = 0; l <= 2; l++)
                    {
                        Add(dp[i][j][k][l], dp[i - 1][j][k][l]);
                        if(j >= a[i])
                            Add(dp[i][j][k][l], dp[i - 1][j - a[i]][k][l]);
                        if(k > 0 && j >= a[i])
                            Add(dp[i][j][k][l], dp[i - 1][j - a[i]][k - 1][l]);
                        if(l > 0)
                            Add(dp[i][j][k][l], dp[i - 1][j][k][l - 1]);
                    }
                }
            }
        }
        LL res = 0;
        for(int i = 0; i <= s; i++)
            res += dp[n][i][2][2];
        printf("%lld
", res * 4 % mod);
    }
    return 0;
}