01背包变形 之 hdu 2126 Buy the souvenirs

/*
题意:
T组测试数据,每组给出 n 和 m,n代表商品数目,m是所拥有的钱,然后给出 n 个商品价格,
求出可以买的最多商品种类数,以及对应的方案数
 
思路:
对于此题,第一个想法便是以价格作为费用,件数1作为价值,01背包即可求出花费 m 所能获得的商品的最大件数 dp[m]。
不过,又要求其方案数,自己试着敲了一下代码WA,后来参考了此篇博客AC掉了。。。
需要建立一个数组 select[j],存储花费 j 元钱买 dp[j] 件商品的方案数。
1) dp[j-arr[i]] + 1 > dp[j]:
	则select[j]是由select[j-arr[i]]决定的,因为当前只不过在花费 j-arr[i] 元钱所能购买的商品中又加入了第i个商品,使之        花费为j,故方案数还是select[j]。
	注意:如果此时 0 == select[j-arr[i]], 则select[j] = 1,因为花费 j 元钱只选第 i 个商品,也是一种方案。
2)dp[j-arr[i]] + 1 == dp[j]:
	如果此时在花费j元时,(选择第i件商品后的商品件数==不选择第i件商品后的商品件数),
	那么 select[j] = select[j-arr[i]] + select[j] (即选/不选第i件商品的两种情况下的方案数的总和)
	注意:如果此时 0 == select[j-arr[i]],则 select[j] = 1 + select[j],解释同上。。
3)dp[j-arr[i]] + 1 < dp[j]:
	可见方案数是由花费 j 元钱,不选第 i 件商品情况下决定的, 故:select[j] = select[j]
*/
#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAX_N = 35, MAX_M = 505;
int n, m, arr[MAX_N], dp[MAX_M], select[MAX_M];

void Solve() {
	memset(dp, 0, sizeof(dp));
	memset(select, 0, sizeof(select));
	for (int i = 1; i <= n; ++i) {
		for (int j = m; j >= arr[i]; --j) {
			if (dp[j - arr[i]] + 1 > dp[j]) {
				dp[j] = dp[j - arr[i]] + 1;
				if (select[j - arr[i]]) select[j] = select[j - arr[i]];
				else select[j] = 1;
			}
			else {
				if (dp[j - arr[i]] + 1 == dp[j]) {
					if (select[j - arr[i]]) select[j] += select[j - arr[i]];
					else select[j] += 1;
				}
			}
		}
	}
	if (select[m]) printf("You have %d selection(s) to buy with %d kind(s) of souvenirs.\n", select[m], dp[m]);
	else printf("Sorry, you can't buy anything.\n");
}

int main()
{
	//freopen("input.txt", "r", stdin);
	//freopen("output.txt", "w", stdout);
	int T;
	scanf("%d", &T);
	while (T--) {
		scanf("%d %d", &n, &m);
		for (int i = 1; i <= n; ++i) {
			scanf("%d", &arr[i]);
		}
		Solve();
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值