Leetcode 938.二叉搜索树的范围和Java&Python

给定二叉搜索树的根结点 root,返回 LR(含)之间的所有结点的值的和。

二叉搜索树保证具有唯一的值。

示例 1:

输入:root = [10,5,15,3,7,null,18], L = 7, R = 15
输出:32

示例 2:

输入:root = [10,5,15,3,7,13,18,1,null,6], L = 6, R = 10
输出:23

提示:

  1. 树中的结点数量最多为 10000 个。
  2. 最终的答案保证小于 2^31

题目涉及到二叉搜索树而不是普通的二叉树,那么在考虑问题的时候,就一定要联想到二叉搜索树的性质。根据题目描述,首先想到的就是遍历整棵树,找到给定区间里的值求和即可,这是普通二叉树的做法,既然是二叉搜索树,那么当我们找到一个节点比L小的时候,那么以这个节点为根的子树就不需要遍历了,因为二叉搜索树的左孩子(子树)小于根,右侧的R也是同样的道理,下面是AC代码:

Java:

public int rangeSumBST(TreeNode root, int L, int R) {
        if (root == null) {
            return 0;
        }
        
        if (root.val < L) {
            return rangeSumBST(root.right, L, R);
        }
        
        if (root.val > R) {
            return rangeSumBST(root.left, L, R);
        }
        
        return root.val + rangeSumBST(root.right, L, R) + rangeSumBST(root.left, L, R);
    }

Python:

def rangeSumBST(self, root, L, R):
        """
        :type root: TreeNode
        :type L: int
        :type R: int
        :rtype: int
        """
        if root is None: return 0

        if root.val < L:
            return self.rangeSumBST(root.right, L, R)

        if root.val > R:
            return self.rangeSumBST(root.left, L, R)

        return root.val + self.rangeSumBST(root.left, L, R) + self.rangeSumBST(root.right, L, R)

猜你喜欢

转载自blog.csdn.net/sinat_33150417/article/details/84326854