PAT 5-3 顺序存储的二叉树的最近的公共祖先问题【二叉树】

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

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

输入格式:

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

题目保证输入正确对应一棵二叉树，且$1\le i,j\le 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

#include<iostream>
#include<cstdio>
#include<math.h>
#include<cstring>
#include<climits>
#include<string>
#include<queue>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<climits>
#include<string>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<list>
#include<algorithm>
using namespace std;
#define rep(i,j,k)for(i=j;i<k;i++)
#define per(i,j,k)for(i=j;i>k;i--)
#define MS(x,y)memset(x,y,sizeof(x))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ll long long
#define abs(x) (x>0?x:-x)
const int INF=0x7ffffff;
const ll MAX=1e18;

const int M=1e4+10;
int tree[M];
int i,j,k,n,m;
int x,y;

int work()
{
   while(1){
    if(tree[x]==0||tree[y]==0)break;
    if(x<y)y/=2;
    else x/=2;
    if(x==y)break;
   }
}

int main()
{
    while(~scanf("%d",&n)){
        for(i=1;i<=n;i++)scanf("%d",&tree[i]);
        scanf("%d%d",&x,&y);
        if(x==y){printf("%d %d\n",x,tree[x]);continue;}
        if(!tree[x])printf("ERROR: T[%d] is NULL\n",x);
        else if(!tree[y])printf("ERROR: T[%d] is NULL\n",y);
        else {
                work();
                printf("%d %d\n",x,tree[x]);
        }
    }
    return 0;
}