二叉树祖先问题

最新推荐文章于 2023-08-02 09:20:35 发布

沧浪之水清兮

最新推荐文章于 2023-08-02 09:20:35 发布

阅读量1k

点赞数

分类专栏： C++

C++ 专栏收录该内容

53 篇文章

订阅专栏

本文介绍了二叉树中寻找两个节点的最近公共祖先的方法。提供了两种不同的实现思路：一种通过判断子节点是否覆盖的方式逐步缩小范围；另一种采用递归方式自底向上查找。这两种方法各有优劣，适用于不同场景。

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

//找到第一个共同祖先

class Solution{
	//若p为root的子孙，则返回true
	bool covers(TreeNode*root, TreeNode*p){
		if(NULL == root)
			return false;
		if(root == p)
			return true;
		
		return covers(root->left,p)||covers(root->right,p);
	}
	
	TreeNode* commonAncestorHelper(TreeNode*root, TreeNode*p, TreeNode*q){
		if(root==NULL)
			return NULL;
		if(root==p || root==q)
			return root;
		
		bool is_p_on_left=covers(root->left,p);
		bool is_q_on_right=covers(root->right,q);
		
		//若p和q不在同一边，则返回root
		if(is_p_on_left != is_q_on_right)
			return root;
		
		//否则就在同一边，遍访那一边
		TreeNode* child_side = is_p_on_left?root->left:root->right;
		return commonAncestorHelper(child_side,p,q);
	}
	
	TreeNode* commonAncetor(TreeNode*root, TreeNode*p, TreeNode*q){
		if(!covers(root,p) || !covers(root,q))	//错误检查
			return NULL;
		
		return commonAncestorHelper(root,p,q);
	}
};

//找到最近祖先
class Solution{
	TreeNode* lowestCommonAnceston(TreeNode* root, TreeNode* A, TreeNode* B){
		if(NULL==root || root==A || root==B)
			return NULL;
		
		TreeNode* left = lowestCommonAnceston(root->left,A,B);
		TreeNode* root = lowestCommonAnceston(root->right,A,B);
		if(left && right)
			return root;
		
		return left?left:right;
		
	}
	
};