7-3 顺序存储的二叉树的最近的公共祖先问题
分数 20
作者 DS课程组
单位 浙江大学
设顺序存储的二叉树中有编号为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
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
C++ (g++)
思路:
用数组存储每个节点的值,因为是顺序二叉树,所以找公共祖先就是使编号较大的数回到上一层(较大的数/2),直到两个数相等。相等的那个编号就是公共祖先的编号。
AC代码:
#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int tree[N],n,i,j;
int num[N];
int main()
{
cin>>n;
for(int i=1;i<=n;i++) cin>>tree[i];
cin>>i>>j;
if(tree[i]==0) cout<<"ERROR: T[" <<i<< "] is NULL";
else if(tree[j]==0) cout<<"ERROR: T[" <<j<< "] is NULL";
else{
if(i==j){
//可能会有i和j编号相同的情况
cout<<i<<" "<<tree[i];
return 0;
}
if(i>j) swap(i,j);
num[i]=1;//可能公共节点在i或j上
while(i!=1){
num[i/2]++;
i/=2;
}
while(j!=1){
if(num[j/2]==1){
cout<<j/2<<" "<<tree[j/2];
break;
}else j/=2;
}
}
}