【算法模板之DFS和BFS】

来源:互联网 发布:js中flag使用方法 编辑:程序博客网 时间:2024/04/29 11:08

DFS:/*该DFS框架以2D坐标范围为例,来体现DFS算法的实现思想。*/#include<cstdio>#include<cstring>#include<cstdlib>using namespace std;const int maxn=100;bool vst[maxn][maxn];    // 访问标记int map[maxn][maxn]; // 坐标范围int dir[4][2]={0,1,0,-1,1,0,-1,0};  // 方向向量,(x,y)周围的四个方向bool CheckEdge(int x,int y) // 边界条件和约束条件的判断{ if(!vst[x][y] && ...)  // 满足条件  return 1; else   // 与约束条件冲突  return 0;}void dfs(int x,int y){ vst[x][y]=1; // 标记该节点被访问过 if(map[x][y]==G) // 出现目标态G {  ......  // 做相应处理  return; } for(int i=0;i<4;i++) {  if(CheckEdge(x+dir[i][0],y+dir[i][1]))  // 按照规则生成下一个节点   dfs(x+dir[i][0],y+dir[i][1]); } return;  // 没有下层搜索节点,回溯}int main(){ ...... return 0;}  ------------------------------------------------------华丽分割线----------------------------------------------------------------------------BFS:/*该框架是2D坐标范围内做BFS设计的,使用STL实现*/#include<cstdio>#include<cstring>#include<queue>#include<algorithm>using namespace std;const int maxn=100;bool vst[maxn][maxn];  // 访问标记int dir[4][2]={0,1,0,-1,1,0,-1,0}; // 方向向量struct State   // BFS队列中的状态数据结构{ int x,y; // 坐标位置 int Step_Counter; // 搜索步数统计器};State a[maxn];bool CheckState(State s)  // 约束条件检验{ if(!vst[s.x][s.y] && ...)  // 满足条件  return 1; else   // 约束条件冲突  return 0;}void bfs(State st){ queue <State> q; // BFS队列 State now,next;  // 定义2个状态,当前和下一个 st.Step_Counter=0; // 计数器清零 q.push(st);   // 入队 vst[st.x][st.y]=1;  // 访问标记 while(!q.empty()) {  now=q.front(); // 取队首元素进行扩展  if(now==G) // 出现目标态,此时为Step_Counter的最小值,可以退出即可  {   ...... // 做相关处理   return;  }  for(int i=0;i<4;i++)  {   next.x=now.x+dir[i][0];  // 按照规则生成下一个状态   next.y=now.y+dir[i][1];   next.Step_Counter=now.Step_Counter+1;  // 计数器加1   if(CheckState(next))  // 如果状态满足约束条件则入队     {    q.push(next);       vst[next.x][next.y]=1;  //访问标记   }  }  q.pop(); // 队首元素出队 } return;}int main(){ ...... return 0;}

原创粉丝点击