找出二叉树中的最低公共父节点(LCA)

最新推荐文章于 2022-10-10 15:22:34 发布

martin_liang

最新推荐文章于 2022-10-10 15:22:34 发布

阅读量1.3k

点赞数

CC 4.0 BY-SA版权

分类专栏： C++/C 算法

本文链接：https://blog.youkuaiyun.com/martin_liang/article/details/8699125

算法同时被 2 个专栏收录

537 篇文章

订阅专栏

C++/C

504 篇文章

订阅专栏

本文介绍了一种寻找二叉树中两个节点的最近公共祖先(LCA)的有效算法。通过递归遍历树结构，该算法能够确定指定节点的最近公共祖先，并使用枚举类型来跟踪查找状态。

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

enum
{
   FOUND_NONE;
   FOUND_ONE;
   FOUND_BOTH;
}
FOUND

Node *FindLCA(Node *pRoot, Node *p1, Node *p2)
{
	Node *pFound = NULL;
	findLCAImpl(pRoot, p1, p2, &pFound)
	
	return pFound;
}


FOUND findLCAImpl(Node *pRoot, Node *p1, Node *p2, Node ** ppFound)
{
    if (!pRoot)
	    return FOUND_NONE;
		
	FOUND find1 = findLCAImpl(pRoot->left, p1, p2, ppFound);
	if (find1 == FOUND_BOTH)
		return find1;
	
	FOUND find2 = findLCAImpl(pRoot->right, p1, p2, ppFound);
	if (find2 == FOUND_BOTH)
		return find2;
		
	if (find1 == FOUND_ONE && find2 == FOUND_ONE)
	{
		*ppFound = pRoot;
		return FOUND_BOTH;
	}
	
	if (find1 == FOUND_ONE || find2 == FOUND_ONE)
	{
		if (pRoot == p1 || pRoot == p2)
		{
			*ppFound = pRoot;
			return FOUND_BOTH;
		}
		else
		{
			return FOUND_ONE;
		}
	}
	
	if (pRoot == p1 || pRoot == p2)
		return FOUND_ONE;
		
	return FOUND_NONE;
}