ACWing.165 小猫爬山 DFS剪枝

ACWing.165 小猫爬山 DFS剪枝

翰翰和达达饲养了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≤108

输入样例:

5 1996
1
2
1994
12
29

输出样例:

2

解题思路

首先分析这道题,有点像贪心,但是用搜索解决要更简单一些,首先需要准备的工作如下:

  1. 一个数组“cat”, 用于存放妹纸小猫的体重

  2. 一个数组“car”, 表示每辆缆车,元素的值表示对应缆车内放置的小猫的体重(载客量)

    为了方便,这里我用容器表示数组,实现对缆车选择的动态模拟,同时可以减少使用的状态变量数目

  3. 变量ans:用于记录最少需要的缆车数目

  4. 变量n, w:按题目要求读入的量

接下来分析解题的思路,本题主要关注的状态量有:已经运送的小猫共多少只,已经租用多少辆缆车,每辆车的当前载重量,深搜的执行分析如下:

对于本体的搜索,参数只需要传入当前的小猫编号即可

  1. 从递归的角度考虑,在每次递归开始时需要对递归终止的条件进行判断,当小猫编号与总数量相等时,对最少需要的缆车数进行更新;
  2. 遍历每辆缆车,寻找能放下小猫的缆车。如果能放入,递归进行下一只小猫,此处记得恢复现场
  3. 对于放不下的情况、第一次搜索(没有创建任何缆车),新建一辆缆车并放入小猫,继续递归搜索,同样记得恢复现场
  4. 剪枝优化:本体的剪枝优化思路相对比较简单,只需要在每次递归开始判断一下当前的方案是否比之前的方案更优,如果不就直接放弃这种方案return掉

代码

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int n = 0, w = 0, ans = 0;
int *cat;
vector<int> car;

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

void dfs(int now){
    if(car.size() >= ans) return;
    if(now == n){
        ans = min(ans, (int)car.size());
        return;
    }
    for(int i = 0; i < car.size(); i++){
        if(car[i] + cat[now] <= w){
            car[i] += cat[now];
            dfs(now + 1);
            car[i] -= cat[now];
        }
    }
    car.push_back(cat[now]);
    dfs(now + 1);
    car.pop_back();
}


int main(){
    cin >> n >> w;
    cat = (int *)malloc(n * sizeof(int));
    for(int i = 0; i < n; i++) scanf("%d", &cat[i]);
    sort(cat, cat + n, cmp);
    ans = n;
    dfs(0);
    cout << ans << endl;
    system("pause");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/yanweiqi1754989931/article/details/109603191