acwing 165. 小猫爬山

题目传送门

题目描述

翰翰和达达饲养了 NN 只小猫,这天,小猫们要去爬山。

经历了千辛万苦,小猫们终于爬上了山顶,但是疲倦的它们再也不想徒步走下山了(呜咕>_<)。

翰翰和达达只好花钱让它们坐索道下山。

索道上的缆车最大承重量为 WW,而 NN 只小猫的重量分别是 C1、C2……CNC1、C2……CN。

当然,每辆缆车上的小猫的重量之和不能超过 WW。

每租用一辆缆车,翰翰和达达就要付 11 美元,所以他们想知道,最少需要付多少美元才能把这 NN 只小猫都运送下山?

输入格式

第 11 行:包含两个用空格隔开的整数,NN 和 WW。

第 2..N+12..N+1 行:每行一个整数,其中第 i+1i+1 行的整数表示第 ii 只小猫的重量 CiCi。

输出格式

输出一个整数,表示最少需要多少美元,也就是最少需要多少辆缆车。

数据范围

1≤N≤181≤N≤18,
1≤Ci≤W≤1081≤Ci≤W≤108

输入样例:

5 1996
1
2
1994
12
29

输出样例:

2

dfs + 剪枝

分析


最重要的剪枝就是 if(k >= ans) return;,没有这个剪枝就会超时!

其次先按小猫重量从大到小排序,先装上重量大的猫也可以减少分支数目

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N = 30;
int n, Wmax;
int ans = 0x3f3f3f3f; // 记录最终结果 
int sum[N]; // sum[i]表示第i辆扯已经装载了多少重量
int w[N]; // 每个小猫的重量 
// 当前考虑第u只小猫,已经使用了k辆车 
void dfs(int u, int k)
{
	if(k >= ans) return; // 重要剪枝!!如果当前搜索到的答案已经比最小的大了,后面不用搜索了
	
	if(u >= n) // 已经装完了所有n只小猫 
	{
		ans = min(ans, k);
		return;
	}
	
	for(int i = 0; i < k; i++)
	{
		// 第i辆车可以装下小猫u 
		if(sum[i] + w[u] <= Wmax)
		{
			sum[i] += w[u];
			dfs(u+1, k);
			sum[i] -= w[u]; // 恢复现场 
		}
	}
	// 新开一辆车
	sum[k] += w[u]; 
	dfs(u+1, k+1);
	sum[k] -= w[u]; 
 } 
 
int main()
{
	scanf("%d%d", &n, &Wmax);
	for(int i = 0; i < n; i++) scanf("%d", &w[i]);
	
	sort(w, w+n);
	reverse(w, w+n); // 从大到小排序,剪枝【可有可无】
	
	dfs(0, 0); // 从第0只小猫开始,使用0辆车 
	
	printf("%d", ans);
	return 0;
}

时间复杂度

参考文章

https://www.acwing.com/solution/content/42165/

posted @ 2022-03-08 10:44  VanHope  阅读(107)  评论(0编辑  收藏  举报