二叉树的LCA
最近公共祖先的两种情况:
- p在左子孙中,q在右子孙中——找到了
- p,q在当前root的一侧
- root碰到 p | q了——找到了
- root还没碰到p|q,这一侧成为新root
dfs遍历左右子树,
- 如果左右子树都能查到p or q,则表明 p 和 q 分别在左右子树中,因此,当前节点即为最近公共祖先;
- 如果左右子树其中一个不为空,则返回非空子树里的较高p or q。
TreeNode* LCA(TreeNode* root, TreeNode* p, TreeeNode* q){
//终止条件,如果找无可找或者已经找到了p,q
if(root==nullptr || root==p ||root == q){
return root;
}
//分别在左右树里去判断是否有p,q
TreeNode* leftr = LCA(root->left,p,q);
TreeNode* rightr = LCA(root->right,p,q);
if(leftr && rightr){
return root; //代表左右两边均有
}
if(left==nullptr)
return rightr;
else
return leftr;
}