LeetCode刷题日记2021-4-27/938. 二叉搜索树的范围和/迭代法

仅供自己学习记录

LeetCode刷题日记2021-4-24

题目描述

给定二叉搜索树的根结点 root，返回值位于范围 [low, high] 之间的所有结点的值的和。

示例 1：

输入：root = [10,5,15,3,7,null,18], low = 7, high = 15
输出：32

示例 2：

输入：root = [10,5,15,3,7,13,18,1,null,6], low = 6, high = 10
输出：23

提示：

• 树中节点数目在范围 [1, 2 * 104] 内
• 1 <= Node.val <= 105
• 1 <= low <= high <= 105
• 所有 Node.val 互不相同

题解思路

将分为一下几种情况来讨论

• 当root节点为空节点时
  直接返回0
• 当root节点的值大于high的时候
  二叉搜索树右节点的值均大于节点的值 因此不需要在考虑右子树的节点值 直接返回左子树
• 当root节点的值小于low的时候
  二叉搜索树左节点的值均小于节点的值 因此不需要考虑左子树的节点值 直接返回右子树
• 当root节点的值位于low和high中间的时候
  返回root节点的值 以及左右子树的范围

题解代码

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)
	return root.val + self.rangeSumBST(root.left,low,high)+self.rangeSumBST(root.right,low,high)