HDU-5800-To My Girlfriend-DP

题意：f(i,j,k,l,m)表示必选a[i],a[j]必不选a[k],a[l],加起来总和为m的个数。求1<=i<=n 1<=j<=n 1<=k<=n 1<=l<=n 1<=m<=s  (i,j,k,l都不相同的) 的总和

思路：显然是一个DP

设dp[i][j][s1][s2] 为选前i个数字，总和为j，确定必选的s1个，确定必不选的s2个，的个数

显然dp[i][j][s1][s2]=dp[i-1][j][s1][s2]+dp[i-1][j][s1][s2-1]+dp[i-1][j-a[i]][s1-1][s2]+dp[i-1][j-a[i]][s1][s2]  (边界处理注意)

dp[0][0][0][0]=1;

由于数组比较大初始化费时间，所以选择滚动数组来压缩空间。

由于i和j   k和l可以互换，最后答案要乘4

#include<bits/stdc++.h>
#define mod 1000000007
using namespace std;
int a[1111];
long long dp[2][1111][3][3];
int main()
{
    int t;
    int n,s;
    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);
        bool sign=0;
        dp[0][0][0][0]=1;
        for(int i=1;i<=n;i++) {
            sign=!sign;
            for(int j=s;j>=0;j--) {
                for(int s1=0;s1<=2;s1++) {
                    for(int s2=0;s2<=2;s2++) {
                        dp[sign][j][s1][s2]=dp[!sign][j][s1][s2]%mod;
                        if(s2>=1) {
                            dp[sign][j][s1][s2]+=dp[!sign][j][s1][s2-1];
                            dp[sign][j][s1][s2]%=mod;
                        }
                        if(j>=a[i]) {
                            dp[sign][j][s1][s2]+=dp[!sign][j-a[i]][s1][s2];
                            dp[sign][j][s1][s2]%=mod;
                        }
                        if(j>=a[i]&&s1>=1) {
                            dp[sign][j][s1][s2]+=dp[!sign][j-a[i]][s1-1][s2];
                            dp[sign][j][s1][s2]%=mod;
                        }
                    }
                }
            }
        }
        long long ans=0;
        for(int i=1;i<=s;i++) {
            ans+=dp[sign][i][2][2];
            ans%=mod;
        }
        ans=(((ans*2)%mod)*2)%mod;
        printf("%lld\n",ans);
    }
    return 0;
}