hdu_5800_To My Girlfriend(变种背包)

题目链接：hdu_5800_To My Girlfriend

题意：

给你n和物品和一个重量m，让你求

题解：

To My Girlfriend

令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)。

由于顺序可以交换，最后结果要*4

 #include<bits/stdc++.h>
 #define mst(a,b) memset(a,b,sizeof(a))
 #define F(i,a,b) for(int i=a;i<=b;++i)
 using namespace std;
 typedef long long ll;

 const int N=,P=1e9+;
 int t,n,s,a[N],dp[N][N][][];

 inline void add(int &a,int b){a+=b;if(a>P)a-=P;}

 int main(){
     scanf("%d",&t);
     while(t--)
     {
         scanf("%d%d",&n,&s);
         F(i,,n)scanf("%d",a+i);
         mst(dp,),dp[][][][]=;
         F(i,,n)F(j,,s)F(ii,,)F(jj,,)
         {
             int *p=&dp[i][j][ii][jj];
             add(*p,dp[i-][j][ii][jj]);//不塞
             if(j>=a[i])add(*p,dp[i-][j-a[i]][ii][jj]);//塞
             if(ii>&&j>=a[i])add(*p,dp[i-][j-a[i]][ii-][jj]);//放入必塞
             if(jj>)add(*p,dp[i-][j][ii][jj-]);//放入必不塞
         }
         ll ans=;
         F(i,,s)ans=(ans+dp[n][i][][])%P;
         printf("%d\n",(ans<<)%P);
     }
     return ;
 }