PTA 顺序存储的二叉树的最近的公共祖先问题

209 阅读1分钟

设顺序存储的二叉树中有编号为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>
using namespace std;
const int  MAX = 1e+4;
int find(int  n, int  m)
{
	while (true)
	{
		if (n > m)
			n /= 2;
		else if (m > n)
			m /= 2;
		else
			break;
	}
	return n;
}
int main()
{
	int n;
	cin >> n;
	int tree[MAX];
	int k = 0 , i , j;
	while (k < n)
		cin >> tree[ k++ ];
	cin >> i >> j;
	int t;
	t = find(i, j);
	if (tree[i - 1] == 0)
	{
		cout << "ERROR: T[" << i << "] is NULL" << endl;
		return 0;
	}
	if (tree[j - 1] == 0)
	{
		cout << "ERROR: T[" << j << "] is NULL" << endl;
		return 0;
	}
	if (i == j)
		cout << i << ' ' << tree[i - 1] << endl;
	if (i != j)
		cout << t << ' ' << tree[t - 1] << endl;
	return 0;
}

提交结果:

2.png