树的直径

定义

树上任意两点间最短距离的最大值。

两种做法

dfs(bfs)

算法流程

$step1:$随便选取一个点$p$，dfs找到离$p$最远的点$q$。

$step2:$对$q$进行同样的dfs，树上所有点离$q$的最短距离则为该树的直径。

时间复杂度：$O(n)$

正确性证明

我们假设离$P$最远的点是$Q$。

$1.$$P$在直径上，易得树的直径为$PQ$

$2.P$不在直径上，我们假设树的直径为$AB$

首先，我们讨论$AB,PQ$有交点的情况，我们设交点为$C$

由于距离$P$最远的点为$Q$，那么我们可得$CQ+BC\le CQ+PC$

即$BC\le PC$

那么我们有$AB=AC+BC\le AC+PC$，与$AB$为直径矛盾，故不成立。

然后是$PQ,AB$无交点的情况。

如图所示，我们设$N$为$PQ$上一点，$M$为$AB$上一点，且有一条路径从$N$到$M$。

我们用和上文相同的方法，得$PQ\ge PN+NM+MB$，即$NQ\ge NM+MB$

所以$NQ+MN\ge MB$，$AB\le AQ$，也与$AB$为树的直径矛盾。

综上所述，我们的算法是正确的。

模板题：poj1985

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int Read(){
	int x=0,f=1;
	char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-')  f=-1;
		ch=getchar();
	}
	while(isdigit(ch)){
		x=(x<<3)+(x<<1)+ch-'0';
		ch=getchar();
	}
	return x*f;
}
int first[200005],nxt[200005],to[200005],w[200005],tot=0;
void Add(int x,int y,int z){
	nxt[++tot]=first[x];
	first[x]=tot;
	to[tot]=y;
	w[tot]=z;
}
int maxn=-1,pos=0;
void dfs(int u,int fa,int dis){
	if(dis>maxn){
		maxn=dis;
		pos=u;
	}
	for(int e=first[u];e;e=nxt[e]){
		int v=to[e];
		if(v==fa)  continue;
		dfs(v,u,dis+w[e]);
	}
}

int main(){
	int n=Read(),m=Read();
	for(int i=1;i<=m;i++){
		int x=Read(),y=Read(),z=Read();
		Add(x,y,z);
		Add(y,x,z);
	}
	dfs(1,0,0);
	maxn=-1;
	dfs(pos,0,0);
	cout<<maxn<<endl;
}

DP

算法流程

$step1:$找到每个点子树中的最长链与次长链$f[i]与g[i]$

$step2:$求出每个点$f[i]+g[i]$的最大值

时间复杂度：$O(n)$

正确性证明

一棵树的直径必定是一条链，那么我们可以想象将一条链折叠，就变成了这样：

那么我们的思路就很清晰了，枚举每一个点，找到其子树中的最长链和次长链（注意不要有重合部分），然后对于所有点求一个最大值即可。

在$DP$是我们采用一个巧妙的方法，用一个$d$数组来记录一个最大值，如果我们搜到的链长大于我们的$d$，那么$d$就成为了我们的次大值，更新$max$即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int Read(){
	int x=0,f=1;
	char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-')  f=-1;
		ch=getchar();
	}
	while(isdigit(ch)){
		x=(x<<3)+(x<<1)+ch-'0';
		ch=getchar();
	}
	return x*f;
}
int first[200005],nxt[200005],to[200005],w[200005],tot=0;
void Add(int x,int y,int z){
	nxt[++tot]=first[x];
	first[x]=tot;
	to[tot]=y;
	w[tot]=z;
}
int d[200005],maxn=-1;
void dp(int u,int fa){
	for(int e=first[u];e;e=nxt[e]){
		int v=to[e];
		if(v==fa)  continue;
		dp(v,u);
		maxn=max(maxn,d[u]+d[v]+w[e]);
		d[u]=max(d[u],d[v]+w[e]);
	}
}
int main(){
	int n=Read(),m=Read();
	for(int i=1;i<=m;i++){
		int x=Read(),y=Read(),z=Read();
		Add(x,y,z);
		Add(y,x,z);
	}
	dp(1,0);
	cout<<maxn<<endl;
}