《剑指offer》 -day19- 搜索与回溯算法（中等）【DFS】

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

题目描述

求 $1+2+...+n$ ，要求不能使用乘除法、for、while、if、else、switch、case 等关键字及条件判断语句（A?B:C）。

限制：

• $1<=n<=10000$

递归 + 逻辑短路 💡

这题可真是有点奇葩，第一反应是想到 暴力 for、等差数列求和。

但是，再一看发现题目中不让用 for 和乘除法。

于是想用递归来写，但是 题目又不让用 if，所以如下递归写法 也行不通。

// 不能有if
class Solution {
    public int sumNums(int n) {
        if (n == 1) return 1;
        return n + sumNums(n - 1);
    }
}

参考了一下 K佬 题解，发现可以使用 逻辑短路 代替 if 实现 递归的终止条件。

代码如下：

class Solution {
    int res = 0;
    
    public int sumNums(int n) {
        // n <= 1时，由于“逻辑短路”，则不会再进行递归，而是直接 return n
        boolean flag = n > 1 && sumNums(n - 1) > 0;
        res += n;
        return res;
    }
}

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

题目描述

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

说明:

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

注意：本题与主站 236 题相同：https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/

DFS（后序）⭐️

思路

• 需要自底向上遍历 二叉树，所以可以采用 后序 遍历的方式。
• 如果遇到某个节点 root 的左、右孩子 分别返回 $p$、$q$，则说明 root 为 所求的 最小公共祖先。

第一次由于把 终止条件写错了，造成错误…

// 第一次的 错误code
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null) return null;
        // 后序
        TreeNode left = lowestCommonAncestor(root.left, p, q); // 左
        TreeNode right = lowestCommonAncestor(root.right, p, q); // 右
        // 根
        if ((left == p && right == q) || (left == q && right == p)) return root;
        if (left == null) return right;
        if (right == null) return left;
        return null;
    }
}

以上代码不要看，正确代码如下：

1. 确定递归参数、返回值

   需要使用后序遍历整颗树，并且需要对“根逻辑”进行判断，所以递归函数需要返回值 TreeNode

   public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q)
   

2. 终止条件

   if (root == null) return null;
   

   注意：如果当前节点 root 为 p、q 中一个，也需要直接返回。（第一次自己写的时候就是这里出错了…）

   if (root == p || root == q) return root;
   

3. 后序遍历左、右子树

4. 根逻辑处理

   • 如果遇到某个节点 root 的左、右孩子 分别返回 $p$、$q$，则说明 root 为 所求的 最小公共祖先，直接 return root
   • 如果左孩子返回 $null$，则说明需要进一步查看 右子树的返回结果，则 return right;
   • 如果右孩子返回 $null$，则说明需要进一步查看 左子树的返回结果，则 return left;
   • 否则，return null;

class Solution {
    public 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); // 右
        // 根
        if ((left == p && right == q) || (left == q && right == p)) return root;
        if (left == null) return right;
        if (right == null) return left;
        return null;
    }
}

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

题目描述

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

说明:

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

注意：本题与主站 235 题相同：https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-search-tree/

DFS 后序（普通二叉树写法）

思路

• 剑指 Offer 68 - II. 二叉树的最近公共祖先 中解法 适用于任意 二叉树，BST 当然也适合。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        // 终止条件
        if (root == null) return root;
        if (root == p || root == q) return root;
        // 后序
        TreeNode left = lowestCommonAncestor(root.left, p, q); // 左
        TreeNode right = lowestCommonAncestor(root.right, p, q); // 右
        // 根
        if (left != null && right != null) return root; // 左右孩子分别为p、q（统一写法）
        else if (left == null && right != null) return right; // 左孩子返回null，则需查看右孩子
        else if (left != null && right == null) return left; // 右孩子返回null，则需查看左孩子
        return null;
    }
}

DFS 前序（BST 写法）⭐️

思路 🤔

• 本题是 BST，所以可以利用 BST 的性质（BST 的中序遍历结果是升序的）
• 所以，从上向下遍历（前序），当 root.val 处于 $[p,q]$ 或 $[q,p]$ 区间中，则说明 root 为二者 最先公共祖先。

class Solution {
    public 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); // 右
        // 根
        if ((left == p && right == q) || (left == q && right == p)) return root;
        if (left == null) return right;
        if (right == null) return left;
        return null;
    }
}