图论学习笔记 - 树的直径

前言

给定一棵树，树中每条边都有一个权值，树中两点之间的距离定义为两点的路径上的边权之和。树中最远的两个节点之间的距离被称为树的直径，连接这两点的路径被称为树的最长链。后者也可称为直径，即直径既是一个数值概念，也可以代指一条路径。

我们一般有两种方法求树的直径，时间复杂度都是 O(N)。

树形DP求树的直径

设 1 号节点为根，“N 个点 N-1 条边的无向图”就可以看作“有根树”。

设 D[x] 表示从节点 x 出发走向以 x 为根的子树，能够到达的最远节点的距离。

设 x 的子节点为 y1，y2，y3，...，yt，edge(x,y) 表示边权，则有：

D[x]=max(D[yi]+edge(x,yi),D[x])

接下来，我们可以考虑对每个节点 x 求出“经过节点 x 的最长链的长度”F[x]

整棵树的直径就是 max(F[x])

经过节点 x 的最长链的长度可以通过四个部分构成：从 yi 到 yi 子树中的最远距离，边(x,yi)，

边(x,yj)，从 yj 到 yi 子树中的最远距离。不妨设 j<i。则有：

F[x]=max(D[yi]+D[yj]+edge(x,yi)+edge(x,yj),F[x])

但我们没有必要使用两层循环来枚举 i,j。在子节点的循环将要枚举到 i 时，D[x] 恰好就保存了从节点 x 出发走向以 yj 为根的子树，能够到达的最远节点的距离。

这个距离就是 max(D[yj]+edge(x,yj))。

所以此时我们先用 D[x]+D[yi]+edge(x,yi) 更新 F[x]，再用 D[yi]+edge(x,yi) 更新 D[x]

void dp(int x){
	v[x]=1;
	for(int i=head[x];i;i=e[i].nxt){
		int y=e[i].to;
		if(v[y]) continue;
		dp(y);
		ans=max(ans,d[x]+d[y]+e[i].val);
		d[x]=max(d[x],d[y]+e[i].val);
	}
}

两次 BFS 求树的直径

通过两次 BFS(DFS) 求出树的直径，还更容易计算出直径上的具体节点。

具体流程：

1. 从任意一个节点出发，通过 BFS 或 DFS 对树进行一次遍历，求出与出发点距离最远的节点，记为 p。
2. 从节点 p 出发，通过 BFS 或 DFS 再进行一次遍历，求出与 p 距离最远的节点，记为 p

从 p 到 q 的路径就是树的一条直径。这是因为 p 一定是直径的一端，否则总能找到一条更长的链，与直径的定义矛盾。既然 p 是直径的一端，那么与 p 距离最远的 q 当然就是直径的另一端。

在第 2 步的遍历过程中，可以记录下来每个点第一次被访问时的前驱节点。最后从 q 递归回到 p，即可得到直径的具体方案