Acwing深度搜索(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
题解:利用dfs存两个参数代表是处理第now只小猫时,并且已经租用cnt个缆车。
dfs函数中对第now只小猫的处理是可以把小猫放在上一个缆车里或者重新购买一个缆车放入;退出函数时是当now==n+1时就不用处理了。对于这种深搜我们可以对其剪枝操作:
(1)假设n只小猫用n个缆车来装,如果深搜过程中要用的缆车超过了n那么直接退出;
(2)由于小猫的重量大小不一,可以按照由大到小的顺序来做,重量小的小猫更容易放置,减少分支;
ac代码:
#include<iostream>
#include<algorithm>
using namespace std;
int n,w;
const int N=20;
int cat[N],cab[N];
int ans;
/*
此处可以加上sort的cmp函数,由大到小排列,减少运算次数
*/
void dfs(int now,int cnt)
{
if(cnt>=ans)
{
return;
}
if(now==n+1)
{
ans=cnt;
return ;
}
for(int i=1;i<=cnt;i++)
{
if(cab[i]+cat[now]<=w)
{
cab[i]+=cat[now];
dfs(now+1,cnt);
cab[i]-=cat[now];
}
}
cab[cnt+1]=cat[now];
dfs(now+1,cnt+1);
cab[cnt+1]=0;
}
int main()
{
cin>>n>>w;
for(int i=1;i<=n;i++)
{
cin>>cat[i];
}
ans=n;
dfs(1,0);
cout<<ans<<endl;
return 0;
}