7-3 顺序存储的二叉树的最近的公共祖先问题 (20分) HBU-DS

7-3 顺序存储的二叉树的最近的公共祖先问题 (20分)

设顺序存储的二叉树中有编号为i和j的两个结点，请设计算法求出它们最近的公共祖先结点的编号和值。

输入格式:

输入第1行给出正整数n（≤1000），即顺序存储的最大容量；第2行给出n个非负整数，其间以空格分隔。其中0代表二叉树中的空结点（如果第1个结点为0，则代表一棵空树）；第3行给出一对结点编号i和j。

题目保证输入正确对应一棵二叉树，且1≤i,j≤n。

输出格式:

如果i或j对应的是空结点，则输出ERROR: T[x] is NULL，其中x是i或j中先发现错误的那个编号；否则在一行中输出编号为i和j的两个结点最近的公共祖先结点的编号和值，其间以1个空格分隔。

输入样例1：

15
4 3 5 1 10 0 7 0 2 0 9 0 0 6 8
11 4

输出样例1：

2 3

输入样例2：

15
4 3 5 1 0 0 7 0 2 0 9 0 0 6 8
12 8

输出样例2：

ERROR: T[12] is NULL

思路

先把两个结点转换到同一层上。再同时分别找他们的双亲。

代码

#include <iostream>
#include <cmath>
using namespace std;
int getCeng(int a)
{
    int tmp = 0;
    int presum = 0;
    while (a > presum)
    {
        tmp += 1;
        presum = pow(2, tmp) - 1;
    }
    return tmp;
}
int main()
{
    int bt[1001] = {0};
    int n;
    cin >> n;
    bt[0] = n;
    for (int i = 1; i <= n; ++i)
        cin >> bt[i];
    int a, b;
    cin >> a >> b;
    if (bt[a] == 0)
    {
        printf("ERROR: T[%d] is NULL", a);
        return 0;
    }
    if (bt[b] == 0)
    {
        printf("ERROR: T[%d] is NULL", b);
        return 0;
    }
    if (a > b)
    {
        int tmp = a;
        a = b;
        b = tmp;
    }
    // b 大 深
    while (getCeng(a) != getCeng(b))
    {
        b = b / 2;
        //b 变成 双亲
    }
    // a b 同层
    while (a != b)
    {
        a /= 2;
        b /= 2;
    }
    cout << a << " " << bt[a];
}