【LeetCode算法学习笔记】二叉树的最近祖先

labuladong算法学习笔记

题目

与树相关，考虑一下递归
二叉树的套路框架：

void traverse(TreeNode root){
    //前序遍历
    traverse(root.left);
    //中序遍历
    traverse(root.right);
    //后序遍历
}

先把框架写出来：

TreeNode lowestCommonAncestor(TreeNode root,TreeNode p,TreeNode q){
    TreeNode left = lowestCommonAncestor(root.left,p,q);
    TreeNode right = lowestCommonAncestor(root.right,p,q);
}

递归问题：
1、 函数是干什么的
2、 函数参数中的变量是什么
3、得到递归结果，应该干什么
即，定义、状态、选择。

解法思路

定义

该函数输入root、p、q三个参数，会返回一个节点：
若root和p、q都在以root为根的树中，则返回q和p的最近公共祖先节点；
若p和q都不在以root为根的树中，则返回null；
若p和q只有一个存在于root为根的树总，则返回不确定节点。

状态

在把[以root为根]转移成[以root的子节点为根]，不断缩小问题规模。

选择

TreeNode lowestCommonAncestor(TreeNode root,TreeNode p,TreeNode q){
    //两种情况的base case
    if(root == null) return null;
    if(root == p || root == q) return root;
    TreeNode left = lowestCommonAncestor(root.left,p,q);
    TreeNode right = lowestCommonAncestor(root.right,p,q);
}

情况1，如果p和q都在以root为根的树中，那么left和right一定分别p和q。
情况2，如果p和q都不在以root为根的树中，直接返回null。
情况3，如果q和p只有一个存在于root为根的树中，函数返回该节点。

TreeNode lowestCommonAncestor(TreeNode root,TreeNode p,TreeNode q){
    if(root == null) return null;
    if(root == p || root == q) return root;
    TreeNode left = lowestCommonAncestor(root.left,p,q);
    TreeNode right = lowestCommonAncestor(root.right,p,q);
    /*后序遍历位置*/
    //情况1
    if(left != null && right != null) return root;
    //情况2
    if(left == null && right == null) return null;
    //情况3
    return left == null ? right ； left;
}

二叉树后序遍历，从下往上，从p和q出发往上走，第一次相交的节点就是这个root。