[AcWing] 小猫爬山

小猫爬山

题目

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

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

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

索道上的缆车最大承重量为W，而N只小猫的重量分别是C1、C2……CN。

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

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

输入格式

第1行：包含两个用空格隔开的整数，N和W。

第2…N+1行：每行一个整数，其中第i+1行的整数表示第i只小猫的重量Ci。

输出格式

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

数据范围

$$\begin{gathered} 1\le N\le 18 \\ 1\le Ci\le W\le 108 \end{gathered}$$

输入样例

5 1996
1
2
1994
12
29

输出样例

2

题解

思路

• DFS 剪枝
• 每次遍历有两种可能：1. 原有车能装下该猫
                                     2.新建一辆车

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 20, MAX = 0x3f3f3f3f;
int a[N], n, w[N], m, k = MAX;

void dfs(int u, int s) {
    if (k <= s) return;   // 最优性剪枝
    if (u == n && k > s) {
        k = s;
        return;
    } 
    
    for (int i = 1; i <= s; i ++) {
        if (a[u] + w[i] <= m) {		// 可行性剪枝	
            w[i] += a[u];
            dfs(u + 1, s);
            w[i] -= a[u];
        }
    }
    
    w[s + 1] = a[u];
    dfs(u + 1, s + 1);
    w[s + 1] = 0;
}
 
int main () {
    cin >> n >> m;
    for (int i = 0; i < n; i ++) cin >> a[i];
    
    // 优化搜索顺序
    sort(a, a + n);
    reverse(a, a + n);
    
    dfs(0, 0);
    
    cout << k;
    return 0;
}