题目描述
给定二叉搜索树的根结点 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
解法
使用有序二叉树的中序遍历可以拿到结果,不过可能存在较多无效的节点遍历。以 rangeSumBST(node,L,R) 函数表示 node 为根节点的二叉树在 [L,R] 之间的范围和,对于节点 node,存在以下三种情况:
若 node 节点值处于 [L,R] 之间,则结果 ret=node.val + rangeSumBST(node.left,L,R) + rangeSumBST(node.right,L,R)
若 node.val<L,则结果 ret=rangeSumBST(node.right,L,R)
若 node.val>L,则结果 ret=rangeSumBST(node.left,L,R)
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def rangeSumBST(self, root: TreeNode, L: int, R: int) -> int:
if not root:
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)