hdu 5800 2016多校6

题意：给出一个序列（1000长度），定义f（i,j,k,l,m) 为a[i],a[j]必取,a[k],a[l]必不取和为m的子集个数。统计任一i,j,k,l,m的和是多少（i!=j!=k!=l）

题解：很容易想的一个dp,转移有一个地方需要考虑一下，就是i和j是有顺序的，不同的顺序要分别统计。

#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;

#define N 1010
#define MOD 1000000007
int dp[N][N][3][3];
int a[N];
void xMod(int &x){
    if(x>=MOD) x-=MOD;
}
int main() {
    //cout << "!!!Hello World!!!" << endl; // prints !!!Hello World!!!
    int T;
    scanf(" %d",&T);
    while(T--){
        memset(dp,0,sizeof(dp));
        int n,s;
        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++){
                int ts = j+a[i];
                for(int e1=0;e1<=2;e1++){
                    for(int e2=0;e2<=2;e2++){
                        if(dp[i-1][j][e1][e2]==0) continue;
                        dp[i][j][e1][e2] += dp[i-1][j][e1][e2];
                        xMod(dp[i][j][e1][e2]);

                        if(e2==0){
                            dp[i][j][e1][e2+1] += dp[i-1][j][e1][e2];
                            xMod(dp[i][j][e1][e2+1]);
                        }
                        if(e2==1){
                            dp[i][j][e1][e2+1] += dp[i-1][j][e1][e2];
                            xMod(dp[i][j][e1][e2+1]);
                            dp[i][j][e1][e2+1] += dp[i-1][j][e1][e2];
                            xMod(dp[i][j][e1][e2+1]);
                        }

                        if(ts<=s){
                            dp[i][ts][e1][e2]+= dp[i-1][j][e1][e2];
                            xMod(dp[i][ts][e1][e2]);

                            if(e1==0){
                                dp[i][ts][e1+1][e2]+= dp[i-1][j][e1][e2];
                                xMod(dp[i][ts][e1+1][e2]);
                            }
                            if(e1==1){
                                dp[i][ts][e1+1][e2]+= dp[i-1][j][e1][e2];
                                xMod(dp[i][ts][e1+1][e2]);

                                dp[i][ts][e1+1][e2]+= dp[i-1][j][e1][e2];
                                xMod(dp[i][ts][e1+1][e2]);
                            }
                        }
                    }
                }
            }
        }
        int ret = 0;
        for(int i=1;i<=s;i++){
            ret = (ret+dp[n][i][2][2])%MOD;
        }
        printf("%d\n",ret);
    }
    return 0;
}