Acwing_165小猫爬山 【DFS之剪枝优化】

题目链接:Acwing_165小猫爬山

这个题目来源于<<算法竞赛进阶指南>>这本书上,这本书很好哟

AC代码:

#include <iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 20;
int n, m, a[N], ans, sum[N];

bool cmp(int a, int b)
{
	return a > b;
}

void dfs(int u, int k)//u表示已经放置小猫数量，k表示小车数量
{
	if (k >= ans)return;
	if (u == n)
	{
		ans = k;
		return;
	}

	for (int i = 0; i < k; i++)//遍历第u只猫可以放入前面k量车么
	{
		if (sum[i] + a[u] <= m)//判断条件,可行性剪枝
		{
			sum[i] += a[u];//车重变化
			dfs(u + 1, k);//放入k不变化，已经放置猫数量+1；
			sum[i] -= a[u];//恢复现场
		}
	}
	sum[k] = a[u];//前面k辆车都放不下了，新开一辆车把猫放进去
	dfs(u + 1, k + 1);//已经放置猫数量+1，车数量+1；
	sum[k] = 0;//恢复现场
}
int main()
{
	cin >> n >> m;
	for (int i = 0; i < n; i++)cin >> a[i];
	ans = n;
	sort(a, a + n, cmp);//小猫重量倒着枚举会减少枚举数量
	dfs(0, 0);
	cout << ans << endl;
	return 0;
}