代码随想录算法训练营第21天 | 二叉搜索树结尾

合集 - 代码随想录算法训练营(24)

1.代码随想录算法训练营第2天 | 数组滑动窗口、螺旋打印2024-07-042.代码随想录算法训练营第1天 | 数组二分法、数组双指针2024-07-033.代码随想录算法训练营第3天 | 链表删除元素、翻转链表2024-07-054.代码随想录算法训练营第4天 | 链表两两交换、删除倒N、链表相交、环形链表2024-07-065.代码随想录算法训练营第6天 | 哈希表的应用、复习快慢指针思想2024-07-086.代码随想录算法训练营第7天 | 哈希表和双指针结合、三数和四数之和2024-07-097.代码随想录算法训练营第8天 | 复习字符串API、双指针2024-07-108.代码随想录算法训练营第9天 | 复习字符串匹配、KMP2024-07-129.代码随想录算法训练营第10天 | 复习队列和栈2024-07-1210.代码随想录算法训练营第11天 | 复习逆波兰表达式求值2024-07-1411.代码随想录算法训练营第13天 | 复习二叉树基础2024-07-1512.代码随想录算法训练营第14天 | 复习二叉树翻转2024-07-1713.代码随想录算法训练营第15天 | 二叉树进阶2024-07-2114.代码随想录算法训练营第16天 | 二叉树更加进阶2024-07-2115.代码随想录算法训练营第17天 | 复习二叉搜索树2024-07-2216.代码随想录算法训练营第18天 | 二叉搜索树进阶2024-07-2317.代码随想录算法训练营第20天 | 二叉搜索树中级2024-07-23

18.代码随想录算法训练营第21天 | 二叉搜索树结尾2024-07-24

19.代码随想录算法训练营第22天 | 开始复习回溯2024-07-2420.代码随想录算法训练营第23天 | 回溯进阶2024-07-2721.代码随想录算法训练营第24天 | 复习组合问题2024-07-2822.代码随想录算法训练营第25天 | 回溯问题完结2024-07-2823.代码随想录算法训练营第27天 | 初入贪心2024-07-3024.代码随想录算法训练营第28天 | 贪心进阶2024-07-30

收起

2024年7月23日

题669. 修剪二叉搜索树
暴力做法是找出所有不符合的节点再一一删除。

import java.util.*;

class Solution {

    ArrayList<Integer> list;

    public TreeNode trimBST(TreeNode root, int low, int high) {
        list = new ArrayList<>();
        digui(root);
        for(int i=0;i<list.size();i++){
            if(list.get(i)<low||list.get(i)>high){
                root = del(root, list.get(i));
            }
            
        }
        return root;
    }

    public void digui(TreeNode root){
        if(root!=null){
            list.add(root.val);
            digui(root.left);
            digui(root.right);
        }
        
    }

    public TreeNode del(TreeNode root, int val){
        if(root==null){
            return null;
        }else if(val>root.val){
            root.right = del(root.right,val);
            return root;
        }else if(val<root.val){
            root.left = del(root.left,val);
            return root;
        }else{
            //左右子树都空
            if(root.left==null && root.right==null){
                return null;
            }else if(root.left!=null && root.right==null){
                return root.left;
            }else if(root.left==null && root.right!=null){
                return root.right;
            }else{
                TreeNode cur = root.right;
                while(cur.left!=null){
                    cur = cur.left;
                }
                cur.left = root.left;
                return root.right;
            }
        }
    }
    
}

优秀做法：依据二叉搜索树的特性，如果左子树都小了，就递归处理右子树。如果右都小了，就递归处理左子树。如果当前节点满足，就递归处理左右。

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if (root == null) return null;
        if (root.val < low) return trimBST(root.right, low, high);
        else if (root.val > high) return trimBST(root.left, low, high);
        root.left = trimBST(root.left, low, high);
        root.right = trimBST(root.right, low, high);
        return root;
    }
}

题108. 将有序数组转换为二叉搜索树
每次寻找中点递归建立即可，类似二分法，注意开闭区间的选择，一般用闭区间方便理解。

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        int left=0, right=nums.length-1;
        int mid = (left+right)/2;
        TreeNode root = new TreeNode(nums[mid]);
        root.left = digui(nums,left,mid-1);
        root.right = digui(nums,mid+1,right);
        return root;
    }

    public TreeNode digui(int[] nums, int left, int right){
        if(left>right){
            return null;
        }else{
            int mid = (left+right)/2;
            TreeNode root = new TreeNode(nums[mid]);
            root.left = digui(nums,left,mid-1);
            root.right = digui(nums,mid+1,right);
            return root;
        }
    }
}

题538. 把二叉搜索树转换为累加树
大于等于，就想到当前节点和右子树了。因此按照右中左的顺序递归处理。

class Solution {

    int sum;

    public TreeNode convertBST(TreeNode root) {
        sum=0;
        digui(root);
        return root;
    }

    public void digui(TreeNode root){
        if(root==null){
            return;
        }else{
            digui(root.right);
            sum+=root.val;
            root.val = sum;
            digui(root.left);
            return;
        }
    }
}