二叉树找最近公共祖先：洛谷 P3884 二叉树问题

题目描述：

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

深度：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间距离。

输入样例：

10                                
1 2                            
1 3                            
2 4
2 5
3 6
3 7
5 8
5 9
6 10
8 6

输出样例：

4
4
8

解题思路：

        1、问到二叉树的宽度和深度，每次有给到一对父子，要建立父亲数组（元素是下标的父亲）、深度数组（某元素的深度）和广度数组（代表某一深度层的宽度），且孩子的深度必定比父亲的深度+1，每插入一个点，对应深度层的宽度+1，根据上述关系可以对输入数据进程处理存储；

        2、问到两个结点间的距离问题，可转换为找最近公共祖先问题，适合采用LCA算法：

首先，在两个结点中，确定一个深度比较深的结点u；

其次，把u回溯到与另一个结点v在同一层；

最后，若提到同一层后u=v，说明v是原输入u的祖先，最近公共祖先为v；若u!=v，把u和v同步回溯，直到u=v，此时即是原输入的最近公共结点

代码实现：

int father[10000]; 
int deep[10000],width[10000];
int n, max_deep = 1, max_width = 1;

int LCA(int u, int v)
{
    if (deep[u] < deep[v])
    {
        swap(u, v);
    }
    while (deep[u] != deep[v])    //把两个结点提到同一层
    {
        u = father[u];
    }
    if (u == v)       //v是u的祖先，公共祖先就是v本身
    {
        return v;
    }
    while (u != v)    //同步溯源，自下而上找到第一个公共祖先
    {
        v = father[v];
        u = father[u];
    }
    return u;
}

int main()
{
    deep[1] = 1;
    width[1] = 1;
    int n = 0;
    int c1 = 0, c2 = 0;
    cin >> n;
    for (int i = 1; i < n; i++)
    {
        cin >> c1 >> c2;
        father[c2] = c1;
        deep[c2] = deep[c1] + 1;
        width[deep[c2]]++;
        if (max_deep < deep[c2])
        {
            max_deep = deep[c2];
        }
    }
    for (int i = 1; i < max_deep; i++)
    {
        if (max_width < width[i])
        {
            max_width = width[i];
        }
    }
    int u, v;
    cin >> u >> v;
    cout << max_deep << endl << max_width << endl; 
    cout << ((deep[u] - deep[LCA(u, v)]) * 2 + deep[v] - deep[LCA(u, v)]) << endl;

    return 0;
}