习题4.5 顺序存储的二叉树的最近的公共祖先问题 (25分)

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

输入格式:

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

题目保证输入正确对应一棵二叉树，且1。

输出格式:

如果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 <stdio.h>
#include <cmath>
int n,s[1001],a,b;
int main() {
    scanf("%d",&n);
    for(int i = 1;i <= n;i ++) {
        scanf("%d",&s[i]);
    }
    scanf("%d%d",&a,&b);
    if(s[a] && s[b]) {
        while((int)log2(a) > (int)log2(b)) a /= 2;
        while((int)log2(b) > (int)log2(a)) b /= 2;
        while(a != b) a /= 2,b /= 2;
        printf("%d %d",a,s[a]);
    }
    else {
        if(!s[a]) printf("ERROR: T[%d] is NULL\n",a);
        else if(!s[b]) printf("ERROR: T[%d] is NULL\n",b);
    }
}