搜索与图论1Accepted7

DFS与BFS

排列数字

#include <iostream>

using namespace std;

const int N = 10;

int n;
int path[N];// 全局数组，保存路径上的值
bool st[N];// 表示当前数是或否被使用过，true表示用过，默认是false

void dfs(int u)// u表示层数，这里u=0表示第0层
{
    if (u == n)// 填满所有位置，输出
    {
        for (int i = 0; i < n; i ++ ) printf("%d ", path[i]);// 输出路径上的值
        puts("");
        return;
    }
    
    for (int i = 1; i <= n; i ++ )// 在每一层选取没有被用过的数，然后进入下一层
    {
        if (!st[i])// 当前数没有被使用过
        {
            path[u] = i;
            st[i] = true;// 标记上使用过
            dfs(u + 1);// 递归下一层
            st[i] = false;// 恢复现场
        }
    }
}

int main()
{
    cin >> n; 
    
    dfs(0);// 从0开始看
    
    return 0;
}

n-皇后问题

#include <iostream>

using namespace std;

const int N = 20;

int n;
char g[N][N];
bool col[N], dg[N], udg[N];// 表示当前列，对角，斜对角是或否被使用过，true表示用过

void dfs(int u)
{
    if (u == n)
    {
        for (int i = 0; i < n; i ++ ) puts(g[i]);
        puts("");
        return;
    }
    
    for (int i = 0; i < n; i ++ )
    {
        if (!col[i] && !dg[u + i] && !udg[n - u + i])// 当前数没有被使用过
        {
            g[u][i] = 'Q';
            col[i] = dg[u + i] = udg[n - u + i] = true;// 标记上使用过
            dfs(u + 1);// 递归下一层
            col[i] = dg[u + i] = udg[n - u + i] = false;// 恢复现场
            g[u][i] = '.';
        }
    }
}

int main()
{
    cin >> n;
    for (int i = 0; i < n; i ++ )
        for (int j = 0; j < n; j ++)
            g[i][j] = '.';
    
    dfs(0);// 从0开始看
    
    return 0;
}

走迷宫

#include <iostream>
#include <cstring>

using namespace std;

typedef pair<int, int> PII;

const int N = 110;

int n, m;
int g[N][N], d[N][N];// g[][]存图，d[][]存距离
PII q[N * N];// 手动模拟队列

int bfs()
{
    int hh = 0, tt = 0;
    q[0] = {0, 0};
    
    memset(d, -1, sizeof d);// 将距离d初始化为-1
    d[0][0] = 0;// 起始点的距离d[0,0] = 0
    
    int dx[4] = {-1, 0 , 1, 0}, dy[4] = {0, 1, 0, -1};// (-1,0)，(0,1)，(1,0)，(0,-1)分别表示当前点的左，下，右，上
    
    while (hh <= tt)
    {
        auto t = q[hh ++ ];// 将队首元素赋值给t
        
        for (int i = 0; i < 4; i ++ )// 站在这个路口，向前后左右四个方向看
        {
            int x = t.first + dx[i];// t.first为当前路口的x轴的值，t.first + dx[i]为向左右走
            int y = t.second + dy[i];
            // 如果如果坐标（x,y）合法（在给定图的范围内）且能走（g[x][y] == 0）且没有走过（d[x][y] == -1）
            if (x >= 0 && x < n && y >= 0 && y < m && g[x][y] == 0 && d[x][y] == -1)// 没有走过且是合法位置则进行搜索 
            {
                d[x][y] = d[t.first][t.second] + 1;
                q[++ tt] = {x, y};
            }
        }
    }
    
    return d[n - 1][m - 1];// 返回终点g[n-1][m-1]的距离
}

int main()
{
    cin >> n >> m;
    
    for (int i = 0; i < n; i ++ )
        for (int j = 0; j < m; j ++ )
            cin >> g[i][j];
            
    cout << bfs() << endl;
    
    return 0;
}

树与图的遍历：拓扑排序

#include <iostream>

using namespace std;

const int N = 100010, M = 2 * N;

int n, m;
int h[N], e[M], ne[M], idx; 
bool st[N];

void dfs(int u)
{
    st[u] = true;// 标记一下，已被搜过了
    
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (!st[j]) = dfs(j);
    }
}

int main()
{
    memset(h, -1, sizeof h);// 初始化
    
    dfs(1);
}

树的重心

#include <iostream>
#include <cstring>

using namespace std;

const int N = 100010, M = 2 * N;

int n, m;
int h[N], e[M], ne[M], idx;// idx存放图中的边
int ans = N;
bool st[N];

void add(int a, int b)// 插入函数
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx ++ ;
}

int dfs(int u)
{
    st[u] = true;
    
    int sum = 1, res = 0;// sum当前连通块中点的数量，res连通块中点的数量的最大值
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (!st[j]) 
        {
            int s = dfs(j);
            res = max(res, s);
            sum += s;
        }
    }
    res = max(res, n -sum);
    ans = min(ans, res);
    return sum;
}

int main()
{
    cin >> n;
    memset(h, -1, sizeof h);// 初始化
    for(int i = 0; i < n - 1; i ++ )
    {
        int a, b;
        cin >> a >> b;
        add(a, b);
        add(b, a);// 无向图插入两次
    }
    
    dfs(1);// 随便从哪个点开始搜
    
    cout << ans << endl;
}

图中点的层次

#include <iostream>
#include <cstring>

using namespace std;

const int N = 100010;

int n, m;
int h[N], e[N], ne[N], idx;
int d[N], q[N];// d[]距离，q[]队列

void add(int a, int b)
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx ++ ;
}

int bfs()
{
    int hh = 0, tt = 0;
    q[0] = 1;
    
    memset(d, -1, sizeof d);
    
    d[1] = 0;
    
    while (hh <= tt)
    {
        int t = q[hh ++ ];
        
        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];// j表示当前点
            if (d[j] == -1)
            {
                d[j] = d[t] + 1;
                q[++ tt] = j;
            }
        }
    }
    return d[n];
}

int main()
{
    cin >> n >> m;
    
    memset(h, -1, sizeof h);// 初始化队头
    
    for (int i = 0; i < m; i ++ )
    {
        int a, b;
        cin >> a >> b;
        add(a, b);
    }
    
    cout << bfs() << endl;
}

拓扑排序

#include <iostream>
#include <cstring>

using namespace std;

const int N = 10010;

int n, m;
int h[N], e[N], ne[N], idx;
int q[N], d[N];//q[]队列 d[]路径

void add(int a, int b)
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx ++ ;
}

bool topsort()
{
    int hh = 0, tt = -1;
    
    for (int i = 1; i <= n; i ++ )
        if (!d[i])
            q[++ tt] = i;
    
    while (hh <= tt)// 队列不为空
    {
        int t = q[hh ++ ];
        
        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            d[j] -- ;
            if (d[j] == 0) q[ ++ tt] = j;
        }
    }
    
    return tt == n - 1;// 判断是否所有点都入队，是有向无环图
}

int main()
{
    cin >> n >> m;
    
    memset(h, -1, sizeof h);
    
    for (int i = 0; i < m; i ++ )
    {
        int a, b;
        cin >> a >> b; 
        add(a, b);
    }
    
    if (topsort())
    {
        for (int i = 0; i < n; i ++ ) printf("%d ", q[i]);
        puts("");
    }
    else puts("-1");
    
    return 0;
}