1. 程式人生 > >LeetCode——938 二叉搜尋樹的範圍和

LeetCode——938 二叉搜尋樹的範圍和

給定二叉搜尋樹的根結點 root,返回 L 和 R(含)之間的所有結點的值的和。

二叉搜尋樹保證具有唯一的值。

示例 1:

輸入:root = [10,5,15,3,7,null,18], L = 7, R = 15
輸出:32

示例 2:

輸入:root = [10,5,15,3,7,13,18,1,null,6], L = 6, R = 10
輸出:23

提示:
樹中的結點數量最多為 10000 個。
最終的答案保證小於 2^31。


思路

二叉搜尋樹使用中序遍歷就可以得到遞增陣列,所以先用中序遍歷得到一個遞增陣列。

  let inorderlist =
[] const inorder = function (root) { if (!root) return; inorder(root.left); inorderlist.push(root.val); inorder(root.right); }

然後迴圈遍歷陣列將 L 和 R 之間的數加起來。

  let sum = 0;
  for (let val of inorderlist) {
    if (val < L) continue;
    else if (val >= L && val <=
R) { sum += val; } else { break; } }

完整程式碼

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} L
 * @param {number} R
 * @return {number}
 */
var rangeSumBST = function(root, L, R) { let inorderlist = [] const inorder = function (root) { if (!root) return; inorder(root.left); inorderlist.push(root.val); inorder(root.right); } inorder(root); let sum = 0; for (let val of inorderlist) { if (val < L) continue; else if (val >= L && val <= R) { sum += val; } else { break; } } return sum; };