基础算法--树与图的深度优先遍历

---

一、前言

从名字上的“深度优先”，我们很容易联想到深度优先搜索（DFS）。顾名思义，树与图的深度优先遍历与BFS，它们的核心思想究其根本都是递归。

1. 基本概念

深度优先遍历是一种用来遍历或者搜索树和图的算法。这种方法就是沿着树的深度搜索树的节点，尽可能地探寻树的分支。
我们可以将深度优先遍历，当作一个执着的人，不到南墙不回头，一条路走到黑，才会返回到分岔口，再一条路走到黑，直到走完所有路。

---

二、代码实现

在树的存储上，我们可以将其当作无向图。树其实是无环连通图，一种比较特殊的图，我们可以用有向图方式存储无向图。

1. 构建树

//使用数组模拟链表
const int N = 1e5 + 10; //假设要求的总共节点数量
const int M = 2 * N; 
int h[N]; //头结点
int e[M]; //相当于链表的val
int ne[M]; //相当于链表的next指针
int idx; //记录时间顺序创建的节点编号

//a 为根，使用数组模拟单链表插入，将b插入a对应的单链表
void add(int a, int b)
{
	e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
//我们可以使用如下方式完成a，b节点间的构建
add(a, b), add(b, a);

2. 树的遍历

int 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); //头结点初始化为 -1
	dfs(1); //从1开始遍历
	return 0;
}

三、例题

1. Acwing 846.树的重心

题目
给定一颗树，树中包含n个结点 (编号1~n) 和n-1条无向边。
请你找到树的重心，并输出将重心删除后，剩余各个连通块中点数的最大值。重心定义:重心是指树中的一个结点，如果将这个点删除后，剩余各个连通块中点数的最大值最小，那么这个节点被称为树的重心。
输入格式
第一行包含整数n，表示树的结点数。
接下来n-1行，每行包含两个整数a和b，表示点a和点b之前存在一条边。
输出格式
输出一个整数m，表示重心的所有的子树中最大的子树的结点数目。
数据范围
1 ≤ n ≤ 1e5
输入样例

9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6

输出样式

4

2. 题解

首先分析题意，题目需要给出了我们一个概念叫树的重心，树的重心：删除一个结点之后，所有连通块中点的数量的最大值，既是自己的最大值也是所有删除其他单个结点的情况下连通块中点的数量的最小值， 那么这个点就是树的重心。

了解完树的重心的，那么题目的思路出来了。我们需要假设删除每个结点的情况下，判断所有情况下删除哪个结点时，连通块中点的数量是最小的。所以思路是需要我们遍历每个结点，这个时候我们就可以使用深度优先遍历。

然后，我们来看看题目给出的样例，如下图所示：

若删除结点1，则剩下的连通块中最大的点的数量是4个，如下图所示：

若删除结点2，则剩下的连通块中最大的点的数量是6个；
若删除结点3，则剩下的连通块中最大的点的数量是7个；
若删除结点4，则剩下的连通块中最大的点的数量是5个。
以此类推，对比其中最小值，答案是4个，即结点1是重心。

题解代码
代码如下：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

const int N = 1e5 + 10;
const int M = 2 * N;
int n;
int ans = N; //存储最小的各种假设去重心之后的最大连通块节点数量，即最终结果
int h[N], e[M], idx;
int ne[M];  //因为用有向图的方式存储无向图
bool st[N]; //标记节点已被遍历
//在链表，a节点之后插入b节点
void add(int a, int b)
{
	e[idx] = a, ne[idx] = h[b], h[b] = idx++;
}
//遍历删除各个假设重心，判断谁是最小
//返回以u为根的子树的节点个数
int dfs(int u)
{
	int res = 0; //存储去重心之后，每个连通块的最大值
	int sum = 1; //去重心的子树节点数量
	st[u] = true; //标记节点已被遍历
	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(res, ans); //记录遍历以来最小值
	return sum;
}
int main()
{
	cin >> n;
	memset(h, -1, sizeof h);

	for (int i = 0; i < n - 1; i++)
	{
		//a与b节点相连
		int a, b;
		cin >> a >> b;
		add(a, b), add(b, a);
	}

	dfs(1); //先遍历 1 根节点

	cout << ans << endl;

	return 0;
}

---

四、总结

递归的自我做题感悟：
1、递归到下一层的时候，灵活运用递归回来的时候前面的条件
2、递归回溯的时候，递归之后的条件是返回给上一层的