1. 程式人生 > >Leetcode 938.二叉搜尋樹的範圍和Java&Python

Leetcode 938.二叉搜尋樹的範圍和Java&Python

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

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

 

示例 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

 

提示:

  1. 樹中的結點數量最多為 10000 個。
  2. 最終的答案保證小於 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)