题目描述
牛牛准备参加学校组织的春游, 出发前牛牛准备往背包里装入一些零食, 牛牛的背包容量为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
输出
8
说明
三种零食总体积小于10,于是每种零食有放入和不放入两种情况,一共有2*2*2 = 8种情况。
每一种零食,有选和不选两种状态,所以可以用递归来做,选与不选,时间复杂度是O(2^n)=O(2^30) = O(8^10) 超时的。
可以通过80%
#include <iostream>
using namespace std;
const int N = 35;
int nums[N];
int res = 0;
void helper(int i, int n, int V)
{
if (i >= n) {
res++;
return;
}
if (V == 0) {
res++;
return;
}
if (V >= nums[i]) {
helper(i + 1, n, V - nums[i]);
}
helper(i + 1, n, V);
}
int main()
{
int n, V;
cin >> n >> V;
for (int i = 0; i < n; i++) {
cin >> nums[i];
}
helper(0, n, V);
cout << res << endl;
}
下面进行剪枝优化,对物品排序,如果发现背包已经满了,直接返回,不再搜索下去。
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 35;
int nums[N];
int res = 0;
void helper(int i, int n, int V)
{
if (i >= n) {
res++;
return;
}
if (V == 0) {
res++;
return;
}
if (V < nums[i]) {
res++;
return;
}
if (V >= nums[i]) {
helper(i + 1, n, V - nums[i]);
}
helper(i + 1, n, V);
}
int main()
{
int n, V;
cin >> n >> V;
for (int i = 0; i < n; i++) {
cin >> nums[i];
}
sort(nums, nums + n);
helper(0, n, V);
cout << res << endl;
}
最后,再加一个判断,如果背包体积大于所以物品体积总和,直接输出2^n
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 35;
int nums[N];
int res = 0;
void helper(int i, int n, int V)
{
if (i >= n) {
res++;
return;
}
if (V == 0) {
res++;
return;
}
if (V < nums[i]) {
res++;
return;
}
if (V >= nums[i]) {
helper(i + 1, n, V - nums[i]);
}
helper(i + 1, n, V);
}
int main()
{
int n, V;
cin >> n >> V;
int temp = 0;
for (int i = 0; i < n; i++) {
cin >> nums[i];
temp += nums[i];
}
sort(nums, nums + n);
if (temp <= V) {
cout << pow(2, n) << endl;
}
else {
helper(0, n, V);
cout << res << endl;
}
}