PTA 数据结构第二版习题4.5 顺序存储的二叉树的最近的公共祖先问题

本文链接：https://blog.youkuaiyun.com/Liannai_/article/details/105081519

1、题目

设顺序存储的二叉树中有编号为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

2、思路

条件：1、顺序存储，易找到父节点 i/2；2、n<=1000,遍历某一节点的所有祖先节点，时间复杂度也不算高。
步骤：1、给较小位置的节点a的所有祖先节点打标记。2、遍历另一个节点b的祖先节点，遇到的第一个a的祖先节点c即是所求的LCA。

3、我的代码

#include <stdio.h>
#include <stdlib.h>
#define MAXN 1005
int tree[MAXN];
int path[MAXN];
int main()
{
	int n,a,b;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&tree[i]);
		path[i]=0;
	}
	scanf("%d%d",&a,&b);
	if(tree[a]==0)
	{
		printf("ERROR: T[%d] is NULL",a);
		return 0;
	}
	if(tree[b]==0)
	{
		printf("ERROR: T[%d] is NULL",b);
		return 0;
	}
	int v1,v2;
	//比较a,b的值，先给较小节点的祖先打标记 
	if(a>b)
	{
		v1=b;
		v2=a;
	}
	else
	{
		v1=a;
		v2=b;
	}
	for(int i=v1;i>=1;i/=2)
	{
		path[i]=1;
	}
	for(int i=v2;i>=1;i/=2)
	{
		if(path[i])
		{
			printf("%d %d",i,tree[i]);
			break;
		}
	 } 
	return 0;
}