題目連結 | 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)