对于深度优先搜索算法,其过程简要来说是对每一个可能的分支路径深入到不能再深入为止,而且每个节点只能访问一次
我最初接触它是李白打酒问题,这个问题是这样的
话说大诗人李白,一生好饮。幸好他从不开车。
一天,他提着酒壶,从家里出来,酒壶中有酒2斗。他边走边唱:
无事街上走,提壶去打酒。
逢店加一倍,遇花喝一斗。
这一路上,他一共遇到店5次,遇到花10次,已知最后一次遇到的是花,他正好把酒喝光了。
请你计算李白遇到店和花的次序,可以把遇店记为a,遇花记为b。则:babaabbabbabbbb 就是合理的次序。像这样的答案一共有多少呢?请你计算出所有可能方案的个数(包含题目给出的)。
这个题是典型的深度优先算法的题,为什么这么说呢,我们去思考这个问题的时候可以发现,每一种情况都是在上一种情况下延伸出来的,也就是说这些情况是在不断递归,层层递进的,而dfs就是基于递归实现的,这样的话我们就可以把这些情况用一棵树来表示,只要去遍历这棵树就能找到答案
回到题目上,李白最后一次会遇到花,然后恰好把酒喝光,这样递归的结束条件就知道了,当遇到店5次,遇到花9次的时候,酒还剩一斗。
递归的最初条件是李白有两斗酒,遇到店和花的次数都是0
递归的过程大概就是这样的一棵树,对于每一次,李白都会有两种情况,遇到店或者遇到花。这样不停的递归下去,直到到达我们设定的递归终点,这时我们就找到了答案
看看代码再理解一下
int ans;
void dfs(int a,int b,int c){ // 分别表示酒 店 花
if(a == 1 && b == 5 && c == 9){
ans++; return ;
}
if(b <= 5){
dfs(2*a,b+1,c); // 遇店
}
if(a > 0 && c <= 9){
dfs(a-1,b,c+1); // 遇花
}
}
int main(){
dfs(2,0,0);
cout << ans << endl;
return 0;
}