算法学习笔记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;
}