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

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

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

输入格式:

输入第1行给出正整数n1000),即顺序存储的最大容量;第2行给出n个非负整数,其间以空格分隔。其中0代表二叉树中的空结点(如果第1个结点为0,则代表一棵空树);第3行给出一对结点编号ij

题目保证输入正确对应一棵二叉树,且1i,jn

输出格式:

如果ij对应的是空结点,则输出ERROR: T[x] is NULL,其中xij中先发现错误的那个编号;否则在一行中输出编号为ij的两个结点最近的公共祖先结点的编号和值,其间以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或者编号×2+1,那反过来找公共祖先也就是一直除以2,直到两个数相等,这时的编号就是公共祖先的位置了,然后输出就可以了,AC代码如下:

#include<iostream>
#define MAX 1010
using namespace std;
void Findfather(int a[],int i,int j)
{
    while(1)
    {
        if(i > j)
            i /= 2;
        if(j > i)
            j /= 2;
        if(j == i)
        {
            cout << j <<" "<< a[j];
            break;
        }
    }
}
int main()
{
    int n,i,j;
    cin >> n;
    int a[MAX];
    for(int k = 1;k <= n; k++)
        cin >> a[k];
    cin >> i >> j;
    if(i == j)
        cout << i <<" "<< a[i];
    else if(a[i] == 0)
        cout <<"ERROR: T["<< i <<"] is NULL";
    else if(a[j] == 0)
        cout <<"ERROR: T["<< j <<"] is NULL";
    else
        Findfather(a,i,j);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/xiao__hei__hei/article/details/80282047