leetcode 669 Trim a Binary Search Tree

最新推荐文章于 2024-09-24 08:39:28 发布

MVincent

最新推荐文章于 2024-09-24 08:39:28 发布

阅读量176

点赞数

文章标签： leetcode java python

本文链接：https://blog.csdn.net/MRxjh/article/details/79082359

版权

Java

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

Python

class Solution:
    def trimBST(self, root, L, R):
        """
        :type root: TreeNode
        :type L: int
        :type R: int
        :rtype: TreeNode
        """
        if not root:
            return None
        
        if root.val<L:
            return self.trimBST(root.right,L,R)
        if root.val>R:
            return self.trimBST(root.left,L,R)
        
        root.left=self.trimBST(root.left,L,R)
        root.right=self.trimBST(root.right,L,R)
        
        return root