【ACWing】165. 小猫爬山

题目地址：

https://www.acwing.com/problem/content/167/

翰翰和达达饲养了$N$只小猫，这天，小猫们要去爬山。经历了千辛万苦，小猫们终于爬上了山顶，但是疲倦的它们再也不想徒步走下山了（呜咕>_<）。翰翰和达达只好花钱让它们坐索道下山。索道上的缆车最大承重量为$W$，而$N$只小猫的重量分别是$C_1,C_2,\dots C_N$。当然，每辆缆车上的小猫的重量之和不能超过$W$。每租用一辆缆车，翰翰和达达就要付$1$美元，所以他们想知道，最少需要付多少美元才能把这$N$只小猫都运送下山？

输入格式：
第$1$行：包含两个用空格隔开的整数，$N$和$W$。第$2,...,N+1$行：每行一个整数，其中第$i+1$行的整数表示第$i$只小猫的重量$C_i$。

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

数据范围：
$1\le N\le 18$
$1\le C_i\le W\le 10^8$

思路是DFS。可以依次枚举第$i$个小猫放在哪辆缆车里，第$i$个小猫可以放在之前已经租用的某个缆车里，也可以放在一个新的缆车里。可以用若干个数组来表示一个方案，例如$(1,2),(3,4)$表示前两个小猫放在第$1$个缆车里，$3$和$4$这两只小猫放在第$2$个缆车里。对于任意一个合法方案，先将缆车内部的小猫编号从小到大排序，然后将缆车按照里面的第一只小猫的编号从小到大排序，显然最终得到的方案和排序之前的方案是等价的（因为这题本质是个组合问题，而不是排列问题），当然也是合法的。可以用数学归纳法证明，任意一个合法方案，都能由上述的搜索顺序得到。所以算法正确。可以加上两个剪枝策略，第一，如果搜到了和已存储的答案相同深度了，则直接返回，不可能得到更优解；第二，可以优先安排最重的那只小猫，这样可以使得每层的分叉数尽可能小，有利于剪枝。递归的出口是放完了$N$只小猫的情形。代码如下：

#include <iostream>
#include <algorithm>
using namespace std;

const int N = 20;
int n, w;
// sum[i]是下标为i的缆车里已经放的总重量
int a[N], sum[N];
int res;

// u是正在安排的小猫下标，k是已经用的缆车个数
void dfs(int u, int k) {
	// 搜不到更优解了，直接退出
    if (k >= res) return;

	// 搜到一个更优解，更新答案并退出
    if (u == n) {
        res = k;
        return;
    }

	// 枚举下标u的小猫放在之前的某个缆车里的情况
    for (int i = 0; i < k; i++) {
    	// 如果放在当前缆车里不会超重，则枚举之
        if (sum[i] + a[u] <= w) {
            sum[i] += a[u];
            dfs(u + 1, k);
            sum[i] -= a[u];
        }
    }
        
	// 枚举下标u的小猫放在一个新的缆车里的情况
    sum[k] = a[u];
    dfs(u + 1, k + 1);
    sum[k] = 0;
}

int main() {
    cin >> n >> w;
    for (int i = 0; i < n; i++) cin >> a[i];

	// 将小猫的重量从大到小排序
    sort(a, a + n, greater<int>());

    res = n;

    dfs(0, 0);
    cout << res << endl;

    return 0;
}

时间复杂度指数级，空间$O(N)$。