LCP 44. 開幕式焰火

WrRan發表於2024-09-12
題目連結 LCP 44. 開幕式焰火
思路 樹的遍歷
題解連結 Go DFS+雜湊表
關鍵點
時間複雜度 \(O(n)\)
空間複雜度 \(O(n)\)

程式碼實現:

class Solution:
    def numColor(self, root: TreeNode) -> int:
        seen = set()
        def dfs(node):
            if node is None:
                return
            seen.add(node.val)
            dfs(node.left)
            dfs(node.right)
        dfs(root)
        return len(seen)

相關文章