AcWing 165. 小猫爬山【DFS，剪枝】

AcWing 165. 小猫爬山

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

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

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

数据范围

1≤N≤18,
1≤Ci≤W≤$10^8$

输入样例：

5 1996
1
2
1994
12
29

输出样例：

2

import java.io.*;
import java.util.Arrays;

public class Main {
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    static PrintWriter out = new PrintWriter(new BufferedWriter(new OutputStreamWriter(System.out)));
    static int N = 20, n, m, res = N;   // res最终的答案，可知最差的情况为：每只猫放到一辆车上，即res=猫的数量
    static int[] cat = new int[N], sum = new int[N];  // cat每只猫的体重，sum第i辆车现有的重量数

    public static void main(String[] args) throws IOException {
        String[] line = br.readLine().split(" ");
        n = Integer.parseInt(line[0]);
        m = Integer.parseInt(line[1]);
        for (int i = 0; i < n; i++) cat[i] = Integer.parseInt(br.readLine());
        Arrays.sort(cat, 0, n);
        for (int i = 0; i < n >> 1; i++) {   // 原数组元素交换实现降序排列
            int temp = cat[i];
            cat[i] = cat[n - 1 - i];
            cat[n - 1 - i] = temp;
        }
        dfs(0, 0);  // 从下标第0只猫中选，目前已选0辆车
        out.println(res);
        br.close();
        out.close();
    }

    public static void dfs(int u, int cnt) {
        if (cnt >= res) return;
        if (u == n) {
            res = cnt;
            return;
        }
        // 目前有1——cnt共有cnt辆车可供选择放置
        for (int i = 0; i < cnt; i++) {
            // 当前的车已有的重量加上想放置的猫的重量要不超过每辆车的重量上限
            if (cat[u] + sum[i] <= m) {
                sum[i] += cat[u];
                dfs(u + 1, cnt);    // 放置下一只猫，可供选择的车依然是cnt辆
                sum[i] -= cat[u];
            }
        }
        // 第二大类情况：新开一辆车
        sum[cnt] = cat[u];
        dfs(u + 1, cnt + 1);
        sum[cnt] = 0;
    }
}