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

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值