设顺序存储的二叉树中有编号为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<stdio.h>
int main(){
int n,q=1,i,j,a[1000];
scanf("%d",&n);
while(q<=n)
scanf("%d",&a[q++]);
scanf("%d %d",&i,&j);
if(a[i]==0){
printf("ERROR: T[%d] is NULL",i);
return 0;
}
else if(a[j]==0){
printf("ERROR: T[%d] is NULL",j);
return 0;
}
while(1){
if(i>j&&i%2!=0)
i=(i-1)/2;
else if(i>j&&i%2==0)
i/=2;
else if(i<j&&j%2!=0)
j=(j-1)/2;
else if(i<j&&j%2==0)
j/=2;
else if(i==j){
printf("%d %d",i,a[i]);
return 0;
}
}
}
总结:
①根据题意根结点编号为1。
②左孩子i的父节点的编号为i/2;右孩子j的父节点的编号为(j-1)/2。
③判断空结点的顺序是按输入i,j的顺序,并非树中结点的顺序。