【力扣日記】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:
    def __init__(self):
        self.res=0
    def rangeSumBST(self, root: TreeNode, L: int, R: int) -> int:
        if not root:return 0
        self.rangeSumBST(root.left,L,R)
        # if root.val>R:return self.res
        if L<=root.val<=R:
            self.res+=root.val
        self.rangeSumBST(root.right,L,R)

        return self.res

思路二：

class Solution:
    def rangeSumBST(self, root: TreeNode, L: int, R: int) -> int:
        if not root :
            return 0
        elif root.val < L:
            return self.rangeSumBST(root.right, L, R)
        elif root.val > R:
            return self.rangeSumBST(root.left, L, R)
        else:
            return self.rangeSumBST(root.left, L, R) + self.rangeSumBST(root.right, L, R) + root.val