LeetCode 题解:938. 二叉搜索树的范围和,栈,JavaScript,详细注释
版权声明: 本文为 InfoQ 作者【Lee Chen】的原创文章。
原文链接:【http://xie.infoq.cn/article/68fbdd639b6f56bfce40fb3d5】。文章转载请联系作者。
本文字数:595 字
阅读完需:约 2 分钟
原题链接:938. 二叉搜索树的范围和
解题思路:
使用中序遍历,即可按顺序获取到二叉搜索树的每个节点。
当节点的值在[low, high]
之间,就进行加和。
/**
* @param {TreeNode} root
* @param {number} low
* @param {number} high
* @return {number}
*/
var rangeSumBST = function (root, low, high) {
let sum = 0 // 缓存结点值之和
let stack = [] // 使用栈辅助遍历
let current = root // 使用节点进行遍历
// 中序遍历二叉树
// 遍历的过程中,可能出现current为空,而stack中有元素,即为遍历到树的最底端。
// 也可能出现current有值而stack为空的情况,如遍历到二叉树的根节点。
while (stack.length || current) {
// 不断向左下遍历节点,同时将所有节点入栈
// 这样就保证了总是从最左端开始遍历节点
while (current) {
stack.push(current)
current = current.left
}
// 从栈中弹出节点,即为当前遍历的节点
// 弹出节点的顺序为从左到右。
const node = stack.pop()
// 处理中序遍历读取到的节点
// 节点的值在[low, high]之间,就进行加和
if (node.val >= low && node.val <= high) {
sum += node.val
}
// 不断向右侧遍历树,如果右侧节点为空,则会从栈中弹出当前子树的根节点
// 这样就形成了从左到右的遍历顺序
current = node.right
}
return sum
}
版权声明: 本文为 InfoQ 作者【Lee Chen】的原创文章。
原文链接:【http://xie.infoq.cn/article/68fbdd639b6f56bfce40fb3d5】。文章转载请联系作者。
还未添加个人签名 2018-08-29 加入
还未添加个人简介
促进软件开发及相关领域知识与创新的传播
评论