二叉树问题

原题P3884 [JLOI2009]二叉树问题
FBI树题解(二叉树基础知识)

题面

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

深度：4(从根节点到字节的最多经过的层数); 宽度：4（同一层最多结点个数）

结点间距离： ⑧→⑥为8 (3×2+2=8)

⑥→⑦为3 （1×2+1=3）

注：结点间距离的定义：由结点向根方向（上行方向）时的边数×2，

与由根向叶结点方向（下行方向）时的边数之和。

输入格式

输入文件第一行为一个整数n(1≤n≤100)，表示二叉树结点个数。接下来的n-1行，表示从结点x到结点y（约定根结点为1），最后一行两个整数u、v，表示求从结点u到结点v的距离。

输出格式

三个数，每个数占一行，依次表示给定二叉树的深度、宽度及结点u到结点v间距离。

解析

由于二叉树是由根结点和左右子树构成, 所以可以创建一个结构, 为每个结点创建一个left变量和right变量, 由于这道题还要求距离, 所以我们不仅要能遍历左右子树, 还要能从一个结点向他的父结点遍历, 所以我们在为每个结点创建一个father变量, 为了做题方便我们还要为结点创建一个floor变量(该结点所在的层数)。遍历树时用一个变量 k代表当前所在的二叉树的层数, 这样深度 h 就是 k 的最大值; 同时用一个数组 bk 来统计每层的结点数目, 即每层的宽度 bk[k](注意: 由于该二叉树可能只存在一个子树, 即在一种极端情况下, 该二叉树成一条链, 此时它的深度就等于结点数, 所以 bk 数组长度要不低于结点数目 n 的最大值)。

以下是我写的详细做法:

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
struct node {
    int left, right, father, floor;
} a[110];
int n, x, y, w, h, d;//n、x、y、与输入中的相同, 最后输入的 u 和 v 由 x 和 y 代替;
//w 代表宽度, h 代表深度, d 代表所求两结点间距离;
int bk[110] = {0};//数组 bk 用来储存第 k 层结点的个数, 即每层的宽度;
bool cmp (int a, int b) {//将每层宽度从大到小排;
    return a > b;
}
void hw(int x, int k) {//遍历树得到深度 h 和宽度每层的宽度 bk[k];
    if (!x) {
        return;
    }
    a[x].floor = k;//标记 x 结点在第 k 层;
    ++bk[k];
    h = max(h, k);//找到最大层数, 即深度;
    hw(a[x].left, k + 1);//遍历左子树;
    hw(a[x].right, k + 1);//遍历右子树;
}
void distance(int x, int y) {//找到 x 到 y 的距离;
//该函数从 x 和 y 结点处向父结点遍历;
    if (x == y) {//当 x 和 y 相等即 x 和 y 遍历到相同父结点, 遍历结束;
        return;
    }
    if (a[x].floor < a[y].floor) {//如果 x 的层数低, 则只有 y 向夫结点遍历, 一下原理相同;
        distance(x, a[y].father);
        ++d;//由共同父结点到 y 结点, 为题中下行方向, 每两个结点间距离为 1;
    }
    else if (a[x].floor > a[y].floor) {
        distance(a[x].father, y);
        d += 2;//由 x 结点到共同父结点, 为题中上行方向, 每两个结点距离为 2;
    }
    else {
        distance(a[x].father, a[y].father);
        d += 3;//两个方向同时遍历, 即距离加 3;
    }

}
int main() {
    cin >> n;
    for (int i = 1; i < n; ++i) {
        cin >> x >> y;
        //如果 x 结点的左子树为空, 优先存入左子树 (也可以优先存入右子树, 不影响解题);
        if (!a[x].left) a[x].left = y;
        else a[x].right = y;
        a[y].father = x;//标记子结点 y 的父亲是 x;
    }
    cin >> x >> y;
    hw(1, 1);
    sort(bk + 1, bk + h + 1, cmp);//将树每层的宽度有大到小排列;
    w = bk[1];//bk[1]是排序后的最大值;
    distance(x, y);
    cout << h << endl << w << endl << d << endl;
    return 0;
}