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

本文链接：https://blog.csdn.net/qq_39225746/article/details/116189465

仅供自己学习记录

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)