BFS DFS模板

转载于https://blog.csdn.net/alalalalalqp/article/details/9155419

BFS模板：

#include<cstdio>

#include<cstring>

#include<queue>

#include<algorithm>

using namespace std;

const int maxn=;

bool vst[maxn][maxn]; // 访问标记

int dir[][]={,,,-,,,-,}; // 方向向量

struct State // BFS 队列中的状态数据结构

{

    int x,y; // 坐标位置

    int Step_Counter; // 搜索步数统计器

};

State a[maxn];

boolCheckState(State s) // 约束条件检验

{

    if(!vst[s.x][s.y] && ...) // 满足条件        

        return ;

    else // 约束条件冲突

    return ;

}

void bfs(State st)

{

    queue <State> q; // BFS 队列

    State now,next; // 定义2 个状态，当前和下一个

    st.Step_Counter=; // 计数器清零

    q.push(st); // 入队    

    vst[st.x][st.y]=; // 访问标记

    while(!q.empty())

    {

        now=q.front(); // 取队首元素进行扩展

        if(now==G) // 出现目标态，此时为Step_Counter 的最小值，可以退出即可

        {

            ...... // 做相关处理

            return;

        }

    for(int i=;i<;i++)

    {

        next.x=now.x+dir[i][]; // 按照规则生成    下一个状态

        next.y=now.y+dir[i][];

        next.Step_Counter=now.Step_Counter+; // 计数器加1

        if(CheckState(next)) // 如果状态满足约束条件则入队

        {

            q.push(next);

            vst[next.x][next.y]=; //访问标记

        }

    }

    q.pop(); // 队首元素出队

    }

 return;

}

int main()

{

......

 return ;

}

DFS模板：

DFS:

/*

该DFS 框架以2D 坐标范围为例，来体现DFS 算法的实现思想。

*/

#include<cstdio>

#include<cstring>

#include<cstdlib>

using namespace std;

const int maxn=;

bool vst[maxn][maxn]; // 访问标记

int map[maxn][maxn]; // 坐标范围

int dir[][]={,,,-,,,-,}; // 方向向量，(x,y)周围的四个方向

bool CheckEdge(int x,int y) // 边界条件和约束条件的判断

{

    if(!vst[x][y] && ...) // 满足条件

        return ;

    else // 与约束条件冲突

        return ;

}

void dfs(int x,int y)

{

    vst[x][y]=; // 标记该节点被访问过

    if(map[x][y]==G) // 出现目标态G

        {

        ...... // 做相应处理

        return;

        }

    for(int i=;i<;i++)

    {

        if(CheckEdge(x+dir[i][],y+dir[i][])) // 按照规则生成下一个节点

            dfs(x+dir[i][],y+dir[i][]);

    }

    return; // 没有下层搜索节点，回溯

}

int main()

{

......

return ;

}

自用，侵删。