hdu 5800 To My Girlfriend

∑i=1n∑j=1n∑k=1n∑l=1n∑m=1sf(i,j,k,l,m)(i,j,k,laredifferent)

i，j为必选数的序号，k，l为必不选数的序号，m是背包体积。

官方题解：

令dp[i][j][s1][s2]表示前i个物品填了j的体积，有s1个物品选为为必选，s2个物品选为必不选的方案数(0<=s1,s2<=2)，则有转移方程dp[i][j][s1][s2] = dp[i - 1][j][s1][s2] + dp[i - 1][j - a[i]][s1 - 1][s2] + dp[i - 1][j][s1][s2 - 1]，边界条件为dp[0][0][0][0] = 1，时间复杂度O(NS*3^2)

我的思路：

对dp[i][j][ii][jj],前i个数中，和达到j，ii个必选，jj个必不选的情况数量
            1、
                第i个数可取可不取:dp[i][j][ii][jj]+=dp[i-1][j][ii][jj]+dp[i-1][j-a[i]][ii][jj];
            2、
                1)必取            dp[i][j][ii][jj]+=dp[i-1][j-a[i]][ii-1][jj];
                2)必不取          dp[i][j][ii][jj]+=dp[i-1][j][ii][jj-1];

因为i和j、k和l之间能替换，所以最后结果要乘4.

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
typedef long long ll;
int a[1005];
int dp[1005][1005][3][3];
const int mod=1e9+7;
int main()
{
   int T;
   scanf("%d",&T);
   int n,s;
   while(T--)
   {
      scanf("%d%d",&n,&s);
      memset(dp,0,sizeof(dp));
      for(int i=1;i<=n;i++)
      {
          scanf("%d",&a[i]);
      }
      dp[0][0][0][0]=1;
      /*
            对dp[i][j][ii][jj],前i个数中，和达到j，ii个必选，jj个必不选的情况数量
            1、
                第i个数可取可不取:dp[i][j][ii][jj]+=dp[i-1][j][ii][jj]+dp[i-1][j-a[i]][ii][jj];

            2、
                1)必取            dp[i][j][ii][jj]+=dp[i-1][j-a[i]][ii-1][jj];
                2)必不取          dp[i][j][ii][jj]+=dp[i-1][j][ii][jj-1];
      */
      for(int i=1;i<=n;i++)///代表第i个数
      {
          for(int j=0;j<=s;j++)///代表和为j的情况
          {
              for(int ii=0;ii<=2;ii++)///第i个数必取的情况
              {
                  for(int jj=0;jj<=2;jj++)///第i个数必不取的情况
                  {
                      dp[i][j][ii][jj]=(dp[i][j][ii][jj]+dp[i-1][j][ii][jj])%mod;
                      if(a[i]<=j)
                        dp[i][j][ii][jj]=(dp[i][j][ii][jj]+dp[i-1][j-a[i]][ii][jj])%mod;
                      if(a[i]<=j&&ii>0)
                        dp[i][j][ii][jj]=(dp[i][j][ii][jj]+dp[i-1][j-a[i]][ii-1][jj])%mod;
                      if(jj>0)
                        dp[i][j][ii][jj]=(dp[i][j][ii][jj]+dp[i-1][j][ii][jj-1])%mod;
                  }
              }
          }
      }
      ll ans=0;
      for(int i=1;i<=s;i++)
      {
         ans=(ans+dp[n][i][2][2])%mod;
      }
        printf("%I64d\n",(4*ans)%mod);
   }
    return 0;
}