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

7-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

---

Code:

#include<stdio.h>
#include<stdlib.h>

int main(void)
{
    int k, n;
    scanf("%d", &n);
    int *s = (int *)malloc(sizeof(int)*n);
    for(k = 0; k < n; k ++) {
        scanf("%d", &s[k]);
    }
    int i, j, flag = -1;
    scanf("%d %d", &i, &j);
    if(s[j-1] == 0) {
        flag = i;
    }
    else if(s[i-1] == 0) {
        flag = i;
    }
    else {
        for(k = 0; k < n; k ++) {
            if(i == j) {
                flag = i;
                break;
            }
            if(i > j) {
                i /= 2;
            }
            else {
                j /= 2;
            }       
        }
    }   
    if(flag == i && flag == j) {
        printf("%d %d", flag, s[flag-1]);
    }
    else {
        printf("ERROR: T[%d] is NULL", flag);
    }

    return 0;
}