Range Sum of BST

첫번째 코드

 

사실 트리에 대해서는 이해하지만 트리를 만드는 코드와 해당 트리를 다루는 방법을 몰라서 GPT를 참고했다.

이렇게 코드를 작성해도 아직 확실한 이해는 잘 모르겠다...

어떻게 트리를 짜야하는지.... 다뤄야 하는지....

 

# 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 dfs(node):
            if not node:
                return 0
            
            if node.val < low:
                return dfs(node.right)
            
            if node.val > high:
                return dfs(node.left)
                
            return node.val + dfs(node.left) + dfs(node.right)

        return dfs(root)

 

트리연습 트리연습!

통과:)

---

링크

https://github.com/ornni/leetcode/tree/main/0938-range-sum-of-bst

leetcode/0938-range-sum-of-bst at main · ornni/leetcode