1151 LCA in a Binary Tree (30 分)

学编程的蒟蒻

于 2022-02-23 22:52:58 发布

阅读量105

点赞数

分类专栏： PAT甲级文章标签： PAT

本文链接：https://blog.youkuaiyun.com/weixin_46309592/article/details/123101398

版权

PAT甲级专栏收录该内容

143 篇文章

订阅专栏

该程序实现了一个寻找二叉树中两个节点最近公共祖先的算法。它使用前序遍历和中序遍历来构造树的映射，并通过深度优先搜索找到最近公共祖先。输入包括树的节点数、节点值、前序遍历和中序遍历的序列，以及要查找的两个节点。如果节点不存在，程序会输出错误信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1151 LCA in a Binary Tree (30 分)

#include <iostream>
#include <map>
using namespace std;
map<int, int> m;
int pre[10010], in[10010], n, M, x, y;
void dfs(int root, int l1, int l2) {
	if(l1 > l2) return;
	int proot = m[pre[root]], px = m[x], py = m[y];
	if(px < proot && py < proot)
		dfs(root + 1, l1, proot - 1);
	else if(px < proot && py > proot || px > proot && py < proot)
		printf("LCA of %d and %d is %d.\n", x, y, pre[root]);
	else if(px > proot && py > proot)
		dfs(root + proot - l1 + 1, proot + 1, l2);
	else if(px == proot)
		printf("%d is an ancestor of %d.\n", x, y);
	else if(py == proot)
		printf("%d is an ancestor of %d.\n", y, x);
}
int main() {
	cin >> M >> n;
	for(int i = 1; i <= n; ++i){
		cin >> in[i];
		m[in[i]] = i;
	}
	for(int i = 1; i <= n; ++i)
		cin >> pre[i];
	while(M--) {
		cin >> x >> y;
		if(m[x] == 0 && m[y] == 0)
			printf("ERROR: %d and %d are not found.\n", x, y);
		else if(m[x] == 0 || m[y] == 0)
			printf("ERROR: %d is not found.\n", m[x] == 0 ? x : y);
		else
			dfs(1, 1, n);
	}
}