Leetcode剑指Offer刷题 - 第十九天

Leetcode剑指Offer刷题指南：

Leetcode剑指Offer刷题-学习计划目录_DEGv587的博客-CSDN博客

剑指 Offer 64. 求1+2+…+n

解法一：等差数列求和公式

class Solution {
    public int sumNums(int n) {
        //Sn = n(n + 1) / 2
        return (int)(Math.pow(n, 2) + n) >> 1;
    }
}

 解法二：通过递归替代循环，通过短路运算符来替代条件判断

class Solution {
    public int sumNums(int n) {
        boolean flag = n > 0 && (n += sumNums(n - 1)) > 0;
        return n;
    }
}

剑指 Offer 68 - I. 二叉搜索树的最近公共祖先

题目关键信息提取：

• 二叉搜索树
• 节点值都是唯一的

那么可以分为三种情况寻找公共祖先：

1.  两个节点 p 和 q 分别在公共祖先 root 的两侧，即 root.val > p.val && root.val < q.val
2. p == root 且 q 在 root 的左右子树中
3. q == root 且 p 在root 的左右子树中

解法一：迭代

循环搜索： 当节点 root 为空时跳出；

1. 当 p, q 都在 root 的 右子树 中，则遍历至 root.right ；
2. 否则，当 p, q 都在 root 的 左子树 中，则遍历至 root.left ；
3. 否则，说明找到了 最近公共祖先 ，跳出。

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        while (root != null) {
            if (root.val < p.val && root.val < q.val) {
                //p，q都在 root 右子树中，root 向右子节点遍历
                root = root.right;
            } else if (root.val > p.val && root.val > q.val) {
                //p，q都在 root 左子树中，root 向左子节点遍历
                root = root.left;
            } else {
                //p，q在root两侧
                break;
            }
        }
        return root;
    }
}

优化版：若可保证 p.val < q.val ，则在循环中可减少判断条件

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (p.val > q.val) {
            //保证 p.val < q.val
            TreeNode tmp = p;
            p = q;
            q = tmp;
        }

        while (root != null) {
            if (root.val < p.val) {
                //比最小都小
                root = root.right;
            } else if (root.val > q.val) {
                //比最大都大
                root = root.left;
            } else {
                break;
            }
        }
        return root;
    }
}

正负号判断 + 迭代

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        while (root != null) {
            if ((p.val - root.val) * (q.val - root.val) <= 0) {
                //在 root 两侧
                return root;
            } else {
                root = p.val > root.val ? root.right : root.left;
            }
        }
        return root;
    }
}

解法二：

递归

1. 当 p, q 都在 root 的 右子树 中，则开启递归 root.right 并返回；
2. 否则，当 p, q 都在 root 的 左子树 中，则开启递归 root.left 并返回；

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root.val < p.val && root.val < q.val) {
            return lowestCommonAncestor(root.right, p, q);
        }
        if (root.val > p.val && root.val > q.val) {
            return lowestCommonAncestor(root.left, p, q);
        }

        return root;
    }
}

剑指 Offer 68 - II. 二叉树的最近公共祖先

题目关键信息提取：

• 所有节点的值都是唯一的
• p、q 为不同节点且均存在于给定的二叉树中

思路：

1. p 和 q 在 root 的子树中，且分列 root 的 异侧（即分别在左、右子树中）；
2. p = root ，且 q 在 root 的左或右子树中；
3. q = root ，且 p 在 root 的左或右子树中；

解法：从底至顶回溯

终止条件：

1. 当越过叶节点，则直接返回 null ；
2. 当 root 等于 p, q ，则直接返回 root

递推工作：

1. 开启递归左子节点，返回值记为 left ；
2. 开启递归右子节点，返回值记为 right

返回值： 根据 left 和 right ，可展开为四种情况

1.当 left 和 right 同时为空 ：说明 root 的左 / 右子树中都不包含 p,q，返回 null

2.当 left 和 right 同时不为空 ：说明 p, q 分列在 root 的 异侧 （分别在 左 / 右子树），因此 root 为最近公共祖先，返回 root

3.当 left 为空 ，right 不为空 ：p,q 都不在 root 的左子树中，直接返回 right。具体可分为两种情况：

• p,q 其中一个在 root 的 右子树 中，此时 right 指向 p（假设为 p ）；
• p,q 两节点都在 root 的 右子树 中，此时的 right 指向 最近公共祖先节点 

4.当 left 不为空 ， right为空 ：与 3. 同理

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null || root == p || root == q) return root;
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);

        if (root.left == null && root.right == null) return null;//可省略
        if (left == null) return right;
        if (right == null) return left;
        return root;
    }
}