题意:找出二叉树中相距最远的两个节点的距离。
-
8 1 2 1 3 1 4 4 5 3 6 6 7 7 8
样例输出
-
6
输入
每个测试点(输入文件)有且仅有一组测试数据。
每组测试数据的第一行为一个整数N,意义如前文所述。
每组测试数据的第2~N行,每行分别描述一根木棍,其中第i+1行为两个整数Ai,Bi,表示第i根木棍连接的两个小球的编号。
对于20%的数据,满足N<=10。
对于50%的数据,满足N<=10^3。
对于100%的数据,满足N<=10^5,1<=Ai<=N, 1<=Bi<=N
小Hi的Tip:那些用数组存储树边的记得要开两倍大小哦!
输出
对于每组测试数据,输出一个整数Ans,表示给出的这棵树中距离最远的两个结点之间相隔的距离。
递归求解,对于每一个树或者子树来说,最长的距离是一棵子树中最长的距离l1(+1)加上另一棵子树最长的距离l2(+1)( maxLen = l1 + l2 + 2)
#include <iostream>
#include <vector>
using namespace std;
vector< vector<int> > g(100005);
int ans = 0;
int dfs(int pre, int cur){
int l1 = -1, l2 = -1; //l1为子树中最大长度,l2为子树中第二大长度
for(int i = 0; i < g[cur].size(); i++){ //枚举每一个子树(子路径)
if(g[cur][i] == pre) continue; //每一个子树(子路径)都只能算一遍,保证不重复走
int l = dfs(cur, g[cur][i]);
if(l > l1) {l2 = l1; l1 = l;} //更新长度
else l2 = max(l, l2);
}
ans = max(ans, l1+l2+2); //对于每一棵子树的根节点都更新一次最终的最大长度
return l1+1;
}
int main()
{
int N;
cin >> N;
int a, b;
for(int i = 1; i < N; i++){
cin >> a >> b;
g[a-1].push_back(b-1);
g[b-1].push_back(a-1);
}
dfs(-1, 0);
cout << ans << endl;
return 0;
}