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];
}