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;}