设顺序存储的二叉树中有编号为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
#include <stdio.h>
#include <stdlib.h>
#define MAX 102400
int Tree[MAX]; //创建树
int main()
{
int n; scanf("%d", &n);
for(int i = 1; i <= n; i++)
{
scanf("%d", &Tree[i]); //向树中输入元素
}
int i, j; scanf("%d %d", &i, &j);
int flag0 = 1;
if(Tree[i] == 0) //ERROR判断
{
flag0 = 0;
printf("ERROR: T[%d] is NULL\n", i);
}
if(Tree[j] == 0)
{
if(flag0 == 1)
printf("ERROR: T[%d] is NULL\n", j);
flag0 = 0;
}
int arr1[100], arr2[100], ArrOneCnt = 0, ArrTwoCnt = 0; //储存路径
while(i != 1)
{
arr1[ArrOneCnt++] = i;
i /= 2;
}
arr1[ArrOneCnt] = 1;
while(j != 1)
{
arr2[ArrTwoCnt++] = j;
j /= 2;
}
arr2[ArrTwoCnt] = j;
int cnt; //编号
int flag = 0;
for(int p = 0; p <= ArrOneCnt; p++)
{
for(int q = 0; q <= ArrTwoCnt; q++)
{
if(arr1[p] == arr2[q]){
cnt = arr1[p];
flag = 1;
break;
}
}
if(flag == 1) break;
}
int cnt_ = Tree[cnt]; //记录值
if(flag0 == 1)
printf("%d %d", cnt, cnt_);
return 0;
}