2019 网易校招01背包问题的递归解法

题目描述

牛牛准备参加学校组织的春游, 出发前牛牛准备往背包里装入一些零食, 牛牛的背包容量为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;
	}
}