题意
n个物品,m元钱,每个物品最多买一次,问最多可以买几件物品,并且输出方案数。
思路
01背包输出路径数量;
用两个一维数组,一个更新最优解,另一个更新最优解对应的路径数;
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=35,M=505;
int cost[N];
int dp[M][2];//dp[i][0]更新最优解,dp[i][1]更新最优解对应的路径数
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=0; i<n; i++)
scanf("%d",&cost[i]);
for(int i=0; i<=m; i++)
dp[i][0]=0,dp[i][1]=1; //起点处初始化为1;
for(int i=0; i<n; i++)
for(int j=m; j>=cost[i]; j--)
{
if(dp[j][0]==dp[j-cost[i]][0]+1) //当前解与最优解相同,路径叠加
dp[j][1]+=dp[j-cost[i]][1];
if(dp[j][0]<dp[j-cost[i]][0]+1) //更新最优解与路径
{
dp[j][0]=dp[j-cost[i]][0]+1;
dp[j][1]=dp[j-cost[i]][1];
}
}
if(dp[m][0]) //最优解不为0
printf("You have %d selection(s) to buy with %d kind(s) of souvenirs.\n",dp[m][1],dp[m][0]);
else
printf("Sorry, you can't buy anything.\n");
}
return 0;
}