数与图的深度优先遍历&树与图的广度优先遍历

树与图的深度优先遍历&树与图的广度优先遍历

树与图的两种遍历方法是基于dfs和bfs思想之上的算法，其思想依然是递归，但是遍历对象变成了无向图，所以在图的构建和搜索过程中多了一些注意事项。

一、树与图的深度优先遍历

深度优先遍历与dfs类似，将树与图中的各条分支给依次遍历一遍。

1、构造

由于是无向图，它构建起来不同于普通的树，首先看代码：

int e[N],ne[N],h[M];//M = 2 * N;
int idx;

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

在输入时：

int a,int b;
cin >> a >> b;
add(a,b),add(b,a);

由于是无向图，所以要add两次。仔细看可以发现，add函数与单链表插入头结点的函数类似，以下是单链表插入头结点的代码：

void head_add(int x)
{
    e[idx] = x,ne[idx] = head,head = idx++;  
}

仅仅是将head换成了h数组。

而在无向图中，h和ne数组发挥的作用是类似但是不同的。h是指向子结点的指针，用于递归部分；而ne搜索到底时跳转回上一个分叉的另一个子结点的指针，用于循环。

2、遍历

dfs函数的代码：

void dfs(int u)
{
    vis[u] = 1;
    
    for(int i = h[u];i != -1;i = ne[i])
    {
        int j = e[i];
        if(vis[j])continue;
        
        int s = dfs(j);
    }
}

在面对不同问题的时候进行一定的修改即可。

3、例题代码

这里以一题为例：846. 树的重心 - AcWing题库

代码如下：

#include<iostream>
#include<cstring>
using namespace std;
const int N = 1e5 + 3, M = N * 2;

int n, e[M], ne[M], h[N], idx;
bool vis[N];
int ans = N;

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

int dfs(int u)
{
	vis[u] = 1;

	int sum = 1, rex = 0;
	for (int i = h[u]; i != -1; i = ne[i])
	{
		int j = e[i];
		if (vis[j]) continue;

		int s = dfs(j);

		rex = max(rex, s);
		sum += s;
	}

	rex = max(rex, n - sum);
	ans = min(ans, rex);

	return sum;
}

int main()
{
	memset(h, -1, sizeof h);
	cin >> n;
	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;
	return 0;
}

二、树与图的广度优先遍历

广度优先遍历的思想与bfs相似，是一层一层地进行搜索。

1、构建

构建方式与深度优先遍历相同

2、遍历

广度优先遍历是对临边的搜索，代码如下：

void bfs()
{
    int hh = 0,tt = 0;
    q[0] = 1;//队列数组
    
    while(hh <= tt)
    {
        int t = q[hh++];
        
        for(int i = h[t];i != -1;i = ne[i])
        {
            int j = e[i];
            if(判断)
                q[++tt] = j;
        }
    }
    
}

它的基本模板如此，想要达到不同的结果，就填入相应的代码即可。

3、例题代码

如题：847. 图中点的层次 - AcWing题库

代码如下：

#include<iostream>
#include<cstring>
using namespace std;
const int N = 1e5 + 3,M = N * 2;

int n,m,idx;
int e[M],ne[M],h[N];
int q[N],d[N];

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];
            
            if(d[j] == -1)
            {
                q[++tt] = j;
                d[j] = d[t] + 1;
            }
        }
    }
    
    return d[n];
}

int main()
{
    scanf("%d%d",&n,&m);
    
    memset(h,-1,sizeof h);
    
    for(int i = 0;i < m;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);
    }
    
    cout << bfs() << endl;
    return 0;
}