CH2201 小猫爬山 DFS

题目链接

http://noi-test.zzstep.com/contest/0x20%E3%80%8C%E6%90%9C%E7%B4%A2%E3%80%8D%E4%BE%8B%E9%A2%98/2201%20%E5%B0%8F%E7%8C%AB%E7%88%AC%E5%B1%B1

分析

设状态为 $(now,cnt)$，$now$ 表示当前要分配第几只猫，$cnt$ 表示已经租用了多少辆缆车；

同时记录下每辆缆车的重量情况，DFS即可。

当前 $cnt$ 若大于等于已求的答案，直接返回；按重量给猫排序，先放重量大的。

AC代码

#include <cstdio>
#include <algorithm>

using namespace std;

inline int read() {
	int num = 0;
	char c = getchar();
	while (c < '0' || c > '9') c = getchar();
	while (c >= '0' && c <= '9')
		num = num * 10 + c - '0', c = getchar();
	return num;
}

const int maxn = 20;

int n, w, a[maxn], c[maxn], ans = 20;

inline int comp(int a, int b) {
	return a > b;
}

void dfs(int now, int cnt) {
	if (now == n + 1) {
		ans = min(ans, cnt);
		return;
	}
	if (cnt >= ans) return;
	for (int i = 1; i <= cnt; ++i)
		if (c[i] + a[now] <= w) {
			c[i] += a[now];
			dfs(now + 1, cnt);
			c[i] -= a[now];
		}
	c[cnt + 1] = a[now];
	dfs(now + 1, cnt + 1);
	c[cnt + 1] = 0;
}

int main() {
	n = read(), w = read();
	for (int i = 1; i <= n; ++i) a[i] = read();
	sort(a + 1, a + n + 1, comp);
	dfs(1, 0);
	printf("%d", ans);
	return 0;
}