树的直径定义:
一棵树中距离最远的两个点之间的距离就是树的直径。
树的直径算法:
1、首先以任意一个点为起点,通过dfs或bfs求出源点到所有点的路径长度。
2、从中选择长度最长的路径到达的点。
3、再次以这个点为源点重复上述步骤,得到的第二个最长的路径连接的两个点就是树的直径。
代码如下
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
const int N = 20005;
int n,m,len[N];
bool vis[N];
vector<int> v[N];
void bfs(){
queue<int> q;
q.push(1);
int cur;
while(!q.empty()){
cur = q.front();
q.pop();
for(int i = 0; i < v[cur].size(); i++){
int x = v[cur][i];
if(vis[x]) continue;
vis[x] = true;
q.push(x);
}
}
q.push(cur);
memset(vis, false, sizeof(vis));
while(!q.empty()){
cur = q.front();
q.pop();
for(int i = 0; i < v[cur].size(); i++){
int x = v[cur][i];
if(vis[x]) continue;
vis[x] = true;
len[x] = len[cur]+1;
q.push(x);
}
}
printf("%d\n", len[cur]);
}
int main()
{
int x;
scanf("%d%d", &n, &m);
for(int i = 2; i <= n+m; i++){
scanf("%d", &x);
v[i].push_back(x);
v[x].push_back(i);
}
bfs();
return 0;
}