数的直径
所谓树的直径指的是树中所有点对之间距离的最大值,我们有两种O(N)的方法可以求解树的直径,实际情况中我们可以视情况选择;
例题
两次BFS
两次遍历的实现在思想上比较简单,我们可以通过反证法证明结论的正确性,由于图的结构是树,所以我们可以使用BFS或者DFS,不影响结论的正确性。
#include<bits/stdc++.h>
using namespace std;
//树的直径两次BFS或者两次DFS实现
const int N=2000010;
const int M=1000010;
int edge[N];
int nest[N];
int last[M];
int cnt=1;
inline void add(int u,int v){
edge[cnt]=v;
nest[cnt]=last[u];
last[u]=cnt;
cnt++;
return;
}
bool vise[M];
int dis=0;
int start;
void DFS(int k,int depth){
if(depth>dis){
dis=depth;
start=k;
}
for(int i=last[k];i;i=nest[i]){
if(vise[edge[i]])continue;
vise[edge[i]]=true;
DFS(edge[i],depth+1);
}
return ;
}
int main(){
int n;
cin>>n;
int u,v;
for(int i=0;i<n;i++){
scanf("%d %d",&u,&v);
add(u,v);
add(v,u);
}
//一次DFS
vise[1]=true;
DFS(1,0);
memset(vise,false,sizeof(vise));
//二次DFS
vise[start]=true;
DFS(start,0);
cout<<dis+1<<endl;
return 0;
}
DP
DP的方式是比较难理解一点,主要思路是我们要理解树的直径的可能的来源,考虑到当我们以点x为根求解每个点的深度,而直径要么经过x要么不经过,不论直积经过那个点,该点一定会出现在DFS的某个点上,并且是该点的两个子树中深度较深的两个点之间产生,我们只需要记录每个节点的深度较深的两个点之间的深度和即可得到经过每个点的路径的最大长度,最后找到所有点里面的最大值,即为树的直径。
#include<bits/stdc++.h>
using namespace std;
const int N=2000010;
int edge[N];
int nest[N];
int val[N];
int last[1000010];
int cnt=1;
inline void add(int u,int v){
nest[cnt]=last[u];
edge[cnt]=v;
last[u]=cnt;
cnt++;
return;
}
int depth[1000010];
int dp[1000010];
bool vise[1000010];
int ans=0;
void DFS(int k){
for(int i=last[k];i;i=nest[i]){
if(vise[edge[i]])continue;
vise[edge[i]]=true;
DFS(edge[i]);
dp[k]=max(dp[k],depth[k]+depth[edge[i]]+1);
depth[k]=max(depth[k],depth[edge[i]]+1);
}
ans=max(ans,dp[k]);
return;
}
int main(){
int n;
scanf("%d",&n);
int u,v;
for(int i=0;i<n-1;i++){
scanf("%d %d",&u,&v);
add(u,v);
add(v,u);
}
vise[1]=true;
DFS(1);
cout<<ans+1<<endl;
return 0;
}