[JLOI2009] 二叉树问题
题目描述
如下图所示的一棵二叉树的深度、宽度及结点间距离分别为:
- 深度: 4 4 4
- 宽度: 4 4 4
- 结点 8 和 6 之间的距离: 8 8 8
- 结点 7 和 6 之间的距离: 3 3 3
其中宽度表示二叉树上同一层最多的结点个数,节点 u , v u, v u,v 之间的距离表示从 u u u 到 v v v 的最短有向路径上向根节点的边数的两倍加上向叶节点的边数。
给定一颗以 1 号结点为根的二叉树,请求出其深度、宽度和两个指定节点 x , y x, y x,y 之间的距离。
输入格式
第一行是一个整数,表示树的结点个数
n
n
n。
接下来
n
−
1
n - 1
n−1 行,每行两个整数
u
,
v
u, v
u,v,表示树上存在一条连接
u
,
v
u, v
u,v 的边。
最后一行有两个整数
x
,
y
x, y
x,y,表示求
x
,
y
x, y
x,y 之间的距离。
输出格式
输出三行,每行一个整数,依次表示二叉树的深度、宽度和 x , y x, y x,y 之间的距离。
样例 #1
样例输入 #1
10
1 2
1 3
2 4
2 5
3 6
3 7
5 8
5 9
6 10
8 6
样例输出 #1
4
4
8
提示
对于全部的测试点,保证 1 ≤ u , v , x , y ≤ n ≤ 100 1 \leq u, v, x, y \leq n \leq 100 1≤u,v,x,y≤n≤100,且给出的是一棵树。保证 u u u 是 v v v 的父结点。
问题链接: P3884 [JLOI2009] 二叉树问题
问题分析: LCA问题,不解释。
参考链接: (略)
题记: (略)
AC的C++语言程序如下:
/* P3884 [JLOI2009] 二叉树问题 */
#include <iostream>
#include <cstring>
using namespace std;
const int N = 100 + 1;
struct Node {
int id;
int father;
int left;
int right;
int depth;
} t[N];
typedef pair<int, int> EDGE;
EDGE e[N];
int n, x, y, w[N];
bool vis[N];
void dfs(int u) {
for (int i = 1; i < n; i++) {
if (e[i].first == u) {
if (t[u].left == 0) t[u].left = e[i].second;
else t[u].right = e[i].second;
t[e[i].second].depth = t[u].depth + 1;
t[e[i].second].father = u;
dfs(e[i].second);
}
}
}
int lca(int u, int v)
{
vis[u] = 1;
while (t[u].father) {
u = t[u].father;
vis[u] = 1;
}
while (!vis[v]) v = t[v].father;
return v;
}
int main()
{
cin >> n;
for (int i = 1; i < n; i++) {
int u, v;
cin >> u >> v;
e[i] = {u, v};
}
cin >> x >> y;
memset(vis, false, sizeof vis);
t[1].depth = 1, t[1].id = 1, t[1].father = 0;
vis[1] = true;
dfs(1);
int maxd = 0;
for (int i = 1; i <= n; i++)
maxd = max(maxd, t[i].depth);
cout << maxd << endl;
memset(w, 0, sizeof w);
int maxw = 0;
for (int i = 1; i <= n; i++)
w[t[i].depth]++;
for (int i = 1; i <= n; i++)
maxw = max(maxw, w[i]);
cout << maxw << endl;
int r = lca(x, y);
cout << (t[x].depth - t[r].depth) * 2 + (t[y].depth - t[r].depth) << endl;
return 0;
}