LEETCODE 938. 二叉搜索树的范围和

LEETCODE 938. 二叉搜索树的范围和

题目描述

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

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

 

示例 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
提示:

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

题目地址
中文版
英文版

代码实现

class Solution(object):
    def rangeSumBST(self, root, L, R):
        """
        :type root: TreeNode
        :type L: int
        :type R: int
        :rtype: int
        """
        if not root:
            return 0
        def recursion(root):
            if not root:
                return 0
            if root.val<L: return recursion(root.right) elif root.val>R:
                return recursion(root.left)
            return recursion(root.left)+recursion(root.right)+root.val
        return recursion(root)
                                                                            
2 次阅读

发表评论

电子邮件地址不会被公开。