Lowest Common Ancestor of a Binary Tree
Example 1:
Input: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
Output: 3
Input: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
Output: 5
出口:在BST中的思路是如果p和q一个大于节点值,一个小于节点值,那么该节点一定就是最近的
祖先节点
若不是则(小于)去左子树(大于)右子树中找,直到找到上述出口所述状态
如果没有BST这个前提,那么就不能用pq分别大于小于作为出口,那么这样的出口在哪呢
出口:节点为空,节点为p,或q
1. 左右子树的返回值都不为null, 由于值唯一左右子树的返回值就是p和q, 此时root为LCA
2. 如果左右子树返回值只有一个不为null, 说明只有p和q存在与左或右子树中, 最先找到的那个节点为LCA
3. 左右子树返回值均为null, p和q均不在树中, 返回null
思路
所有的递归的返回值有4种可能性,null、p、q、公共祖先
两个节点p,q分为两种情况:
p和q在相同子树中
p和q在不同子树中
从根节点遍历,递归向左右子树查询节点信息
递归终止条件:如果当前节点为空或等于p或q,则返回当前节点
递归遍历左右子树,如果左右子树查到节点都不为空,
则表明p和q分别在左右子树中,因此,当前节点即为最近公共祖先
如果左右子树其中一个不为空,则返回非空节点。
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root==null||root==p||root==q){
return root;
}//首先对root进行特殊考虑
TreeNode left=lowestCommonAncestor(root.left,p,q);
TreeNode right=lowestCommonAncestor(root.right,p,q);
//root是公共节点的情况(此时left right分别是p、q)
if(left!=null&&right!=null){
return root;
}
if(left!=null){
return left;
}
if(right!=null){
return right;
}
return null;
}
}
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (!root || root == p || root == q) return root;
TreeNode *left = lowestCommonAncestor(root->left, p, q);
TreeNode *right = lowestCommonAncestor(root->right, p, q);
if (left && right) return root;//分在左右子树的状况
return left ? left : right;//分在同一颗子树的情况
}
};