/* 题意: 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; }
01背包变形 之 hdu 2126 Buy the souvenirs
最新推荐文章于 2020-07-31 20:08:40 发布