.
序言
開啟python刷題時代,主要也是為了面試。
.
題目
Given the root node of a binary search tree and two integers low and high, return the sum of values of all nodes with a value in the inclusive range [low, high].
Example 1:
- Input: root = [10,5,15,3,7,null,18], low = 7, high = 15
- Output: 32
- Explanation: Nodes 7, 10, and 15 are in the range [7, 15]. 7 + 10 + 15 = 32.
Example 2:
- Input: root = [10,5,15,3,7,13,18,1,null,6], low = 6, high = 10
- Output: 23
- Explanation: Nodes 6, 7, and 10 are in the range [6, 10]. 6 + 7 + 10 = 23.
Constraints:
- The number of nodes in the tree is in the range [1, 2 * 104].
- 1 <= Node.val <= 105
- 1 <= low <= high <= 105
- All Node.val are unique.
.
思路
簡單的樹遍歷問題,可以用DFS也可以用BFS。
都是第一次用python寫,決定先用DFS試試。
DFS遍歷樹節點,判斷每個節點的val是否大于low并且小于high。
.
代碼
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:def rangeSumBST(self, root: Optional[TreeNode], low: int, high: int) -> int:def sumVal(root):return root.val if low <= root.val <= high else 0def searchTree(root):if root is None:return 0return sumVal(root) + searchTree(root.left) + searchTree(root.right)return searchTree(root)
.
后序
python和C++的語法差異比我想象中的大一些。
目前感覺確實要容易很多。
繼續加油!