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

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

思路：

建个数组存储每个结点的值，找公共祖先的方法，就是不断使编号较大的数回到上一层(较大的编号/2)，直到两个数相等，相等的那个编号就是公共祖先的编号。

更新的AC代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 1010;
int tree[N], n, i, j;
int num[N];
int main(){
    cin >> n;
    for(int i = 1;i <= n;i++)   cin >> tree[i];
    cin >> i >> j;
    if(tree[i] == 0)    cout << "ERROR: T[" << i << "] is NULL";
    else if(tree[j] == 0)   cout << "ERROR: T[" << j << "] is NULL";
    else{
        if(i == j){	//可能会有i,j编号相同的情况
            cout << i << " " << tree[i];
            return 0;
        }
        if(i > j)   swap(i, j);
        num[i] = 1;	//可能公共结点在i或j上
        while(i != 1){
            num[i/2]++;
            i /= 2;
        }
        while(j != 1){
            if(num[j/2] == 1){
                cout << j/2 << " " << tree[j/2];
                break;   
            }else   j /= 2;
        }
    }
}

AC代码：

#include<bits/stdc++.h>
using namespace std;
int find(int n,int m){
	while(1){
		if(n > m)
			n = n/2;
		else if(m > n)
			m = m/2;
		if(n == m)	//编号相等时就是找到公共结点的时刻
		return n;
	}
}
int main(){
	int tree[1001];
	memset(tree,0,sizeof(tree));	//将tree数组置零
	int final;
	int n;
	cin >> n;
	for(int i = 1;i <= n;i++){
		cin >> tree[i];
	}
	int a,b;
	cin >> a >> b;
	final = find(a,b);
	if(a == b){		//编号相等直接将其输出
		cout << a << " " << tree[a] << endl;
	}
	else if(tree[a] == 0){
		printf("ERROR: T[%d] is NULL",a);
	}
	else if(tree[b] == 0){
		printf("ERROR: T[%d] is NULL",b);
	}
	else{
		cout << final << " " <<tree[final] << endl;
	}
}