leetcode 刷题_day23_二叉树09_BST 修剪 & 转换

669. 修剪二叉搜索树

思路

根据 BST 的特性（有序），对 BST 进行遍历，对每个节点的值与 val 进行判断，最后遇到 null 就是需要插入的地方

解题方法

描述你的解题方法

复杂度

• 时间复杂度:

添加时间复杂度, 示例： $O(n)$

• 空间复杂度:

添加空间复杂度, 示例： $O(n)$

Code

  /**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    // 根据 BST 特性，递归遍历
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) {
            TreeNode node = new TreeNode(val);
            return node;
        }

        if (root.val > val) {
            root.left = insertIntoBST(root.left, val);
        }
        if (root.val < val) {
            root.right = insertIntoBST(root.right, val);
        }

        return root;
    }

    
}

108.将有序数组转换为二叉搜索树

思路

前序递归遍历，每次递归返回删除了节点的树/子树根节点

解题方法

处理当前节点时，分为五种情况

1. 没找到待删除节点

待删除节点为当前节点

2. 待删除节点为叶子节点，直接删除当前节点
3. 待删除节点左子节点不空，右子节点为空，返回左子节点
4. 待删除节点左子节点为空，右子节点不空，返回右子节点
5. 带删除节点左右子节点都不为空
   5.1 返回右子节点，将左子树整体迁移到右子树最左叶子节点上，作为其左子节点 --> 左子树小于右子树的任意节点

复杂度

• 时间复杂度:

添加时间复杂度, 示例： $O(n)$

• 空间复杂度:

添加空间复杂度, 示例： $O(n)$

Code

  /**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    // 返回删除了节点的树/子树根节点
    public TreeNode deleteNode(TreeNode root, int key) {
        // 终止条件 --> 找到要删除的节点 --> 五种情况 --> 执行删除逻辑
        // 1. 没找到待删除节点
        if (root == null) {
            return null;
        }
        // 待删除节点为当前节点
        if (root.val == key) {
            // 2. 待删除节点为叶子节点，直接删除当前节点
            if (root.left == null && root.right == null) {
                return null;
            }
            // 3. 待删除节点左子节点不空，右子节点为空，返回左子节点
            else if (root.left != null && root.right == null) {
                return root.left;
            }
            // 4. 待删除节点左子节点为空，右子节点不空，返回右子节点
            else if (root.left == null && root.left != null) {
                return root.right;
            }
            // 5. 带删除节点左右子节点都不为空
            else {
                // 返回右子节点，将左子树整体迁移到右子树最左叶子节点上，作为其左子节点 --> 左子树小于右子树的任意节点
                TreeNode cur = root.right;
                while (cur.left != null) {
                    cur = cur.left;
                }
                cur.left = root.left;
                return root.right;
            }
        }

        if (key > root.val) {
            root.right = deleteNode(root.right, key);
        }
        else if (key < root.val) {
            root.left = deleteNode(root.left, key);
        }

        return root;
    }
}

538.把二叉搜索树转换为累加树

思路

寻找分割点，分割点作为当前节点，然后递归左区间和右区间
注意区间边界条件在每次递归的一致性

解题方法

描述你的解题方法

复杂度

• 时间复杂度:

添加时间复杂度, 示例： $O(n)$

• 空间复杂度:

添加空间复杂度, 示例： $O(n)$

Code

  /**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    // 寻找分割点，分割点作为当前节点，然后递归左区间和右区间
    public TreeNode sortedArrayToBST(int[] nums) {
        TreeNode root = traversal(nums, 0, nums.length - 1);
        return root;
    }

    // 左闭右闭
    private TreeNode traversal(int[] nums, int start, int end) {
        if (start > end) {
            return null;
        }
        // 分割点构建当前节点
        int mid = (start + end) / 2;
        TreeNode node = new TreeNode(nums[mid]);
        // 左右区间继续递归
        node.left = traversal(nums, start, mid - 1);
        node.right = traversal(nums, mid + 1, end);

        return node;
    }
}