牛牛准备参加学校组织的春游, 出发前牛牛准备往背包里装入一些零食, 牛牛的背包容量为w。
牛牛家里一共有n袋零食, 第i袋零食体积为v[i]。
牛牛想知道在总体积不超过背包容量的情况下,他一共有多少种零食放法(总体积为0也算一种放法)。
输入描述:
输入包括两行
第一行为两个正整数n和w(1 <= n <= 30, 1 <= w <= 2 * 10^9),表示零食的数量和背包的容量。
第二行n个正整数v[i](0 <= v[i] <= 10^9),表示每袋零食的体积。
输出描述:
输出一个正整数, 表示牛牛一共有多少种零食放法。
输入例子1:
3 10
1 2 4
输出例子1:
8
例子说明1:
三种零食总体积小于10,于是每种零食有放入和不放入两种情况,一共有2*2*2 = 8种情况。
递归超时
#include <iostream>
#include <vector>
using namespace std;
int reSolver(int id, int left, vector<int>& v) {
if (id == v.size()) return 1;
if (left - v[id] > 0)
return reSolver(id + 1, left - v[id], v)
+ reSolver(id + 1, left, v);
return reSolver(id + 1, left, v);
}
int main() {
int nSnack = 0;
int vBag = 0;
cin >> nSnack >> vBag;
vector<int> v(nSnack, 0);
for (int i = 0; i < nSnack; ++i) {
cin >> v[i];
}
cout << reSolver(0, vBag, v);
return 0;
}
动态规划 ,注意要用 long, 因为牛牛的背包巨大
#include <iostream>
#include <map>
#include <vector>
using namespace std;
struct cmp_key
{
bool operator()(const long &a, const long &b) const
{
return a > b;
}
};
int main() {
long vBag = 0;
int nSnack = 0;
cin >> nSnack >> vBag;
vector<long> v(nSnack, 0);
// v, num
map<long, int, cmp_key> mp;
mp[0] = 1;
for (int i = 0; i != nSnack; ++i) {
cin >> v[i];
}
for (int i = 0; i != nSnack; ++i) {
for (auto it = mp.begin(); it != mp.end(); ++it) {
long key = it->first;
int val = it->second;
long newKey = key + v[i];
if (newKey > vBag) continue;
mp[newKey] += mp[key];
}
}
int cnt = 0;
for (auto it = begin(mp); it != end(mp); ++it) {
cnt += it->second;
}
cout << cnt << endl;
}