Acwing 165 小猫爬山【DFS剪枝与优化】

最新推荐文章于 2024-08-14 16:27:10 发布

耐、

最新推荐文章于 2024-08-14 16:27:10 发布

阅读量80

点赞数

文章标签：算法 dfs 剪枝

本文链接：https://blog.csdn.net/qq_43964401/article/details/108833836

版权

翰翰和达达饲养了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、策略：枚举每只小猫放在哪个缆车上和新加一个缆车
2、搜索顺序：重量从大到小的去枚举每只小猫，即优先搜索分支数较少的
3、可行性剪枝：如果当前的小猫 c[u] + sum[i] > w ，就直接continue;
4：最优性剪枝：当cnt >= ans ，就直接return

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 20;

int n,w;
int c[N];
int sum[N];
int ans = N;

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

int main(){
    scanf("%d%d",&n,&w);
    for(int i = 0 ; i < n ;  i++) scanf("%d",&c[i]);
    
    sort(c,c+n);
    reverse(c,c+n);
    
    dfs(0,0);
    
    cout << ans << endl;
    return 0;
}