Leetcode 938.二叉搜尋樹的範圍和Java&Python
阿新 • • 發佈:2018-11-22
給定二叉搜尋樹的根結點 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
。
題目涉及到二叉搜尋樹而不是普通的二叉樹,那麼在考慮問題的時候,就一定要聯想到二叉搜尋樹的性質。根據題目描述,首先想到的就是遍歷整棵樹,找到給定區間裡的值求和即可,這是普通二叉樹的做法,既然是二叉搜尋樹,那麼當我們找到一個節點比L小的時候,那麼以這個節點為根的子樹就不需要遍歷了,因為二叉搜尋樹的左孩子(子樹)小於根,右側的R也是同樣的道理,下面是AC程式碼:
Java:
public int rangeSumBST(TreeNode root, int L, int R) { if (root == null) { return 0; } if (root.val < L) { return rangeSumBST(root.right, L, R); } if (root.val > R) { return rangeSumBST(root.left, L, R); } return root.val + rangeSumBST(root.right, L, R) + rangeSumBST(root.left, L, R); }
Python:
def rangeSumBST(self, root, L, R): """ :type root: TreeNode :type L: int :type R: int :rtype: int """ if root is None: return 0 if root.val < L: return self.rangeSumBST(root.right, L, R) if root.val > R: return self.rangeSumBST(root.left, L, R) return root.val + self.rangeSumBST(root.left, L, R) + self.rangeSumBST(root.right, L, R)