算法学习笔记11-DFS深度优先搜索

DFS

深度优先搜索是树遍历算法的一种，每次遍历下一结点，直到符合条件或者遍历到叶节点。

具体看两道题目吧：

题目一 24点游戏

问题描述：给出4个1-10的数字，通过加减乘除，得到数字为24就算胜利

输入：

4个1-10的数字。[数字允许重复，测试用例保证无异常数字]

输出：

true or false

#coding=utf-8
from itertools import permutations
def dfs(a,n,i):
flag=False
if n==24:
return True
else:
if n>24 or i>=4:
return False
else:
return dfs(a,n+a[i],i+1)or\
dfs(a,n-a[i],i+1)or\
dfs(a,n*a[i],i+1)or\
dfs(a,n/a[i],i+1)#只要加减乘除中有一种正确即可。
a=map(int,raw_input().split())
flag=False
for i in permutations(a):#求数组的全排列
if dfs(i,i[0],1):
print 'true'
flag=True
break
if not flag:
print 'false'

题目二 求和

输入两个整数 n 和 m，从数列1，2，3…….n 中随意取几个数,使其和等于 m ,要求将其中所有的可能组合列出来

#include<iostream>
#include<vector>
using namespace std;
void dfs(int n,int m,vector<int>&v,int beg)
{
if(m==0)
{
for(int i=0;i<v.size();i++)
{
if(i==0)
cout<<v[i];
else
cout<<" "<<v[i];
}
cout<<endl;
}
for(int i=beg;i<=n&&i<=m;i++)
{
v.push_back(i);
dfs(n,m-i,v,i+1);
v.pop_back();
}
}
int main()
{
int n,m;
cin>>n>>m;
vector<int>v;
dfs(n,m,v,1);
return 0;
}