题目描述:给定二叉搜索树的根结点 root
,返回值位于范围 [low, high]
之间的所有结点的值的和。
我的思路:既然是二叉搜索树,那么如果当前节点如果大于high,那么就应该去递归它的左孩子,如果小于low,就去递归右孩子。若是在范围内,则返回当前节点值+递归左孩子的返回值+递归右孩子的返回值。若当前节点为空,返回0.
代码:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def rangeSumBST(self, root: TreeNode, low: int, high: int) -> int:
if not root:return 0
if root.val>high:
return self.rangeSumBST(root.left, low, high)
if root.val<low:
return self.rangeSumBST(root.right, low, high)
if low<=root.val<=high:
return root.val+self.rangeSumBST(root.left, low, high)+self.rangeSumBST(root.right, low, high)
结果:
官方解答:
思路差不多
class Solution(object):
def rangeSumBST(self, root, L, R):
def dfs(node):
if node:
if L <= node.val <= R:
self.ans += node.val
if L < node.val:
dfs(node.left)
if node.val < R:
dfs(node.right)
self.ans = 0
dfs(root)
return self.ans