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

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

设顺序存储的二叉树中有编号为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

解

我感觉这道题有两个点需要注意一下：

• 顺序存储，有效值编号（下标）从1开始，即根节点是第一个结点，不是第0个结点
• 第a个结点双亲的编号是 a/2（a 除以2 取整）,这个是二叉树五大性质之一

AC代码（C++）

#include<bits/stdc++.h>

using namespace std;

int fun(int a, int b);

int main()
{
    int a[1004], n, i, j;
    cin >> n;
    for(i = 1; i <= n; i++)
        cin >> a[i];
    cin >> i >> j;
    if(a[i] == 0)
        printf("ERROR: T[%d] is NULL\n", i);
    else if(a[j] == 0)
        printf("ERROR: T[%d] is NULL\n", j);
    else
    {
        int index = fun(i, j);
        printf("%d %d\n", index, a[index]);
    }
    return 0;
}
int fun(int a, int b)
{
    if(a == b)
        return a;
    else if(a > b)
        fun(a / 2, b);
    else if(a < b)
        fun(a, b / 2);
}