题意: 有n种纪念品,每种纪念品x元,你有m元,求最多能买多少纪念品及对应方案数。
思路: 若不求方案数则为一个普通的01背包问题,每个纪念品的重量为其价格,价值为1;加上求方案数则要在原来的基础上加多一维记录对应的方案数。
采用二维的滚动数组求解( dp(2, maxn) ):
第一维:dp(0,j)即为01背包最大纪念品数求解,表示容量为j的背包最多能买多少件纪念品。
第二维:dp(1,j)为方案数求解,表示容量为j的背包装最多件纪念品的方案数。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=550;//最大体积;
//最大数量为30;
int souvenir[35];
int dp[2][maxn];//二维dp,第一维为最大数量求解,第二维为方案数求解;
int main()
{
int i,j;
int n,m;
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)
{
scanf("%d",&souvenir[i]);
}
for(j=0;j<=m;j++)
{
dp[0][j]=0;//背包每个单位初始化为0;
dp[1][j]=1;//方案数每个单位初始化为1(即啥也不买也是一种方案);
}
for(i=1;i<=n;i++)//滚动数组求解;
{
for(j=m;j>=souvenir[i];j--)
{
if(dp[0][j]<dp[0][j-souvenir[i]]+1)
{
dp[0][j]=dp[0][j-souvenir[i]]+1;
dp[1][j]=dp[1][j-souvenir[i]];//最大能装的纪念品数改变,对应的方案数改变;
}
else if(dp[0][j]==dp[0][j-souvenir[i]]+1)//装或不装该物品得到相同结果时,对应的方案数相加.
{
dp[1][j]+=dp[1][j-souvenir[i]];
}
}
}
if(dp[0][m]==0) printf("Sorry, you can't buy anything.\n");
else printf("You have %d selection(s) to buy with %d kind(s) of souvenirs.\n",dp[1][m],dp[0][m]);
}
return 0;
}