16 多校 6 - H - To My Girlfriend （HDU - 5800）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5800

题目大意就是在n个数中取任意个它们的和为m且其中含有ai和aj不含ak和al，i、j、k、l互不相同且在范围[1,n]内，m属于[1,s]，就是背包问题，第i个物品有取和不取两种取法，若取，则其有两种可能，一种是必须取的，一种是不是必须取的也不是不能取的，不取时同理，有不能取和介于取与不取之间的。
dp[i][j][k][l] = dp[i-1][j][k][l] + dp[i-1][j][k][l-1] + dp[i-1][j-a[i]][k][l] + dp[i-1][j-a[i]][k-1][l]。

#include<iostream>
#include<string.h>
#include<stdio.h>
using namespace std;
int t,n,s,a[1010], dp[1010][1010][3][3];
int mod = 1000000000+7;
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        memset(dp,0,sizeof dp);
        scanf("%d%d",&n,&s);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        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++)
                    {
                        int num = 0;
                        num = (num + dp[i-1][j][k][l])%mod;
                        if(l)
                            num = (num + dp[i-1][j][k][l-1])%mod;
                        if(j>=a[i])
                        {
                            num = (num + dp[i-1][j-a[i]][k][l])%mod;
                            if(k)
                                num = (num + dp[i-1][j-a[i]][k-1][l])%mod;
                        }
                        dp[i][j][k][l] = num;
                    }
                }
            }
        }
        long long ans = 0;
        for(int i=1;i<=s;i++)
            ans = (ans + dp[n][i][2][2]) % mod;
        printf("%lld\n",ans * 4 % (long long)mod);
    }
    return 0;
}