深度优先搜索算法

来源:互联网 发布:mp4下载软件 编辑:程序博客网 时间:2024/05/29 03:44
今天我们来复习一下万能的搜索算法之深度优先搜索算法。
深度优先搜索算法顾名思义就是按照树的延伸不停的往下搜索,直到树的尽头之后再一步一步的回溯回来。
好吧,我们直接问你一个问题,给你一个数N,让你输出从1到这个树的全排列,你会怎么写,会不会想到去用若干个for循环?
好吧,你肯定错了,其实他考的就是我们的深度优先搜索算法:
#include<iostream>
using namespace std;
#include<stdio.h>
int card[10];
int flag[10];
int step;
int N;
int dfs(int step)
{
if(step == N+1)
{

                for(int index = 1;index <= N;index++)
{
cout<<card[index]<<" ";
}
cout<<endl;
return 0;
}
for(int i =1;i<= N;i++)
{
if(flag[i] == 0)
{
card[step] = i;
flag[i] = 1;
dfs(step+1);
flag[i] = 0;
}
}
}
int _tmain(int argc, _TCHAR* argv[])
{
cin>>N;
dfs(1);
return 0;
}
这就是最最简单而有直接的搜索算法。
那么我们在延伸一下,从这些全排列中找出XXXX +YYYY = ZZZZ的排列怎么办?其实只需要稍加修改即可:


#include<iostream>
using namespace std;
#include<stdio.h>
int card[10];
int flag[10];
int step;
int N;
int dfs(int step)
{
if(step == N+1)
{
if((card[1]*100+card[2]*10 + card[3] + card[4]*100+card[5]*10+ card[6]) ==( card[7]*100+card[8]*10+card[9]))
{
  for(int index = 1;index <= N;index++)
  {
cout<<card[index]<<" ";
  }
cout<<endl;
return 0;
}

}
for(int i =1;i<= N;i++)
{
if(flag[i] == 0)
{
card[step] = i;
flag[i] = 1;
dfs(step+1);
flag[i] = 0;
}
}
}
int _tmain(int argc, _TCHAR* argv[])
{
cin>>N;
dfs(1);
return 0;
}

0 0