欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

leetcode 669. 修剪二叉搜索树 数据结构

程序员文章站 2024-03-23 09:31:46
...

给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。

//递归:
class Solution {
    public TreeNode trimBST(TreeNode root, int L, int R) {
        return dfs(root,L,R);
    }
    public TreeNode dfs(TreeNode cur,int L,int R){
        if(cur==null){ //为null直接返回null
            return null;
        }
        if(cur.val<L){ //如果小于的话直接遍历右边结点
            return dfs(cur.right,L,R);
        }else if(cur.val>R){ //如果大于直接遍历左边结点
            return dfs(cur.left,L,R);
        }
        cur.left=dfs(cur.left,L,R); //遍历左右
        cur.right=dfs(cur.right,L,R);
        return cur;
    }
}