HDU 5800 To My Girlfriend（计数DP）

[题目链接]

[题意]
有n个物品，每个物品的价值为$a_i$，设$f(i,j,k,l,m)$表示从n个物品中取若干个，满足第i，j个物品必取，第k，l个物品必不取，价值总和为m的方案数，求

$$\sum_{i=1}^n\sum_{j=1}^n\sum_{k=1}^n\sum_{l=1}^n\sum_{m=1}^s{f(i,j,k,l,m)} （i,j,k,l各不相同）$$

[分析]
看到一堆西格玛的公式就一脸懵逼
题解的状态表示十分巧妙…
设f[i][j][a][b]表示前i个物品中必选a个，必不选b个，价值总和为j的方案数
a和b只需0~2即可，且i只与i-1有关，可以滚动数组优化空间，空间是O(s)
每个状态只有4个转移策略：
1. 必取f[i+1][j+x][a+1][b]
2. 取f[i+1][j+x][a][b]
3. 必不取f[i+1][j][a][b+1]
4. 不取f[i+1][j][a][b]
所以时间是O(n*s)
初始f[0][0][0][0] = 1 , 转移方法就是累加
最后，答案即为$4*\sum_{m=1}^s{f[n][m][2][2]} \\$
为何要*4？因为统计时虽然保证了每个物品不会被取2次（或不取2次），但没考虑必取（和必不取）的2个物品的位置关系，所以分别*2，最终答案*4

[代码]

#include <bits/stdc++.h>
using namespace std ;
const int N = 1000 + 5 ;
const int mod = 1e9 + 7 ;
typedef long long LL ;

int T , n , s , f[2][N+N][4][4] ;

void add( int &x , int y )
{
    if( (x += y) >= mod )
        x -= mod ;
}

int main()
{
    scanf( "%d" , &T ) ;
    while( T-- )
    {
        scanf( "%d%d" , &n , &s ) ;
        memset(f,0,sizeof(f)) ;
        int t = 1 ;
        f[0][0][0][0] = 1 ;
        for( int i = 1 ; i <= n ; i++ )
        {
            int x ;
            scanf( "%d" , &x ) ;
            for( int j = 0 ; j <= s ; j++ )
                for( int a = 0 ; a <= 2 ; a++ )
                    for( int b = 0 ; b <= 2 ; b++ )
                    {
                        int pre = f[t^1][j][a][b] ;
                        if( pre == 0 ) continue ;
                        add(f[t][j][a][b],pre) ;
                        add(f[t][j][a][b+1],pre) ;
                        add(f[t][j+x][a][b],pre) ;
                        add(f[t][j+x][a+1][b],pre) ;
                    }
            t ^= 1 ;
            memset(f[t],0,sizeof(f[t])) ;
        }
        int sum = 0 ;
        for( int i = 1 ; i <= s ; i++ )
            add(sum,f[t^1][i][2][2]) ;
        printf( "%d\n" , sum*4LL%mod ) ;
    }
    return 0 ;
}