BFS/DFS 模板 代码

来源:互联网 发布:mac修改dock 图标 编辑:程序博客网 时间:2024/05/13 11:53
#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];boolCheckState(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; // 计数器加1if(CheckState(next)) // 如果状态满足约束条件则入队{q.push(next);vst[next.x][next.y]=1; //访问标记}}q.pop(); // 队首元素出队} return;}int main(){...... return 0;}



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;}


原创粉丝点击