设顺序存储的二叉树中有编号为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
#include<cstdio>
#include<iostream>
#include<string>
#include<cstdlib>
using namespace std;
const int maxn = 1e3 + 10;
int tree[maxn];
int fun(int u, int v) {
if (u == v) return u;
else if (u < v) return fun(u, v / 2);
else if (u > v) return fun(u / 2, v);
}
int main() {
int n; cin >> n;
for (int i = 1; i <= n; i++) cin >> tree[i];
int u, v;
cin >> u>> v;
if (tree[u] == 0) printf("ERROR: T[%d] is NULL\n", u);
else if (tree[v] == 0) printf("ERROR: T[%d] is NULL\n", v);
else {
int index = fun(u, v);
printf("%d %d", index,tree[index]);
}
}