设顺序存储的二叉树中有编号为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 findFather(int n,int m);
int main()
{
int binTree[1011]={'\0'};//顺序树
int N;
int result=0;
scanf("%d",&N);//输入树的容量
int i=0;
for(i=0;i<N;i++)
{
scanf("%d",&binTree[i]);
}
int num1=1,num2=0;
scanf("%d %d",&num1,&num2);
result=findFather(num1,num2);
if(num1==num2)
printf("%d %d\n",num1,binTree[num1-1]);//结点重合
else if(binTree[num1-1]==0)//节点不存在
printf("ERROR: T[%d] is NULL\n",num1);
else if(binTree[num2-1]==0)
printf("ERROR: T[%d] is NULL\n",num2);
else
printf("%d %d",result,binTree[result-1]);
}
int findFather(int n,int m)
{
while(1)
{
if(n>m)
n=n/2;
else
m=m/2;
if(n==m)
break;
}
return n;
}
解析:
结果有四种可能:
1.两结点重合
2.一个结点是空节点
3.另一个结点是空节点
4.排除以上情况,有公共祖先
也就对应着这些代码:
if(num1==num2)
printf("%d %d\n",num1,binTree[num1-1]);//结点重合
else if(binTree[num1-1]==0)//节点不存在
printf("ERROR: T[%d] is NULL\n",num1);
else if(binTree[num2-1]==0)
printf("ERROR: T[%d] is NULL\n",num2);
else
printf("%d %d",result,binTree[result-1]);
前三种情况不需要进行复杂的判断,只需要判断输入的数据即可,只有情况4才会用到函数findFather的结果。