来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/range-sum-of-bst 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

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

解答

二叉搜索树查找范围和的优化:1. 如果比当前节点小了,就不再往左子节点走;2. 如果比当前节点大了,就不再往右子节点走

  1. /**
  2. * Definition for a binary tree node.
  3. * function TreeNode(val, left, right) {
  4. * this.val = (val===undefined ? 0 : val)
  5. * this.left = (left===undefined ? null : left)
  6. * this.right = (right===undefined ? null : right)
  7. * }
  8. */
  9. /**
  10. * @param {TreeNode} root
  11. * @param {number} low
  12. * @param {number} high
  13. * @return {number}
  14. */
  15. var rangeSumBST = function(root, low, high) {
  16. let sum = 0;
  17. function traverse (node) {
  18. if (!node) return;
  19. const val = node.val;
  20. val >= low && traverse(node.left);
  21. if (low <= val && val <= high) {
  22. sum += val;
  23. }
  24. val <= high && traverse(node.right);
  25. }
  26. traverse(root);
  27. return sum;
  28. };