P3884 [JLOI2009] 二叉树问题【LCA】

[JLOI2009] 二叉树问题

题目描述

如下图所示的一棵二叉树的深度、宽度及结点间距离分别为：

• 深度：$4$
• 宽度：$4$
• 结点 8 和 6 之间的距离：$8$
• 结点 7 和 6 之间的距离：$3$

其中宽度表示二叉树上同一层最多的结点个数，节点 $u,v$ 之间的距离表示从 $u$ 到 $v$ 的最短有向路径上向根节点的边数的两倍加上向叶节点的边数。

给定一颗以 1 号结点为根的二叉树，请求出其深度、宽度和两个指定节点 $x,y$ 之间的距离。

输入格式

第一行是一个整数，表示树的结点个数 $n$。
接下来 $n-1$ 行，每行两个整数 $u,v$，表示树上存在一条连接 $u,v$ 的边。
最后一行有两个整数 $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\le u,v,x,y\le n\le 100$，且给出的是一棵树。保证 $u$ 是 $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;
}