[99클럽] 12일차 TIL

leetcode의 dfs 문제를 풀어보았습니다.

 

https://leetcode.com/problems/deepest-leaves-sum/submissions/1273992637/

 

아직은 클래스로 구현하는것이 어렵지만 여러 문제를 많이 풀어봐야 할 것 같습니다.

 

from collections import defaultdict

class Solution:
    def deepestLeavesSum(self, root: Optional[TreeNode]) -> int:
        nodelevel = defaultdict(list)

        def dfs(currentNode : TreeNode, level: int):
            if not nodelevel[level]:
                nodelevel[level] = []
            
            nodelevel[level].append(currentNode.val)

            if currentNode.left:
                dfs(currentNode.left, level + 1)
            if currentNode.right:
                dfs(currentNode.right, level + 1)
        dfs(root, 0)

        maxLevel = max(nodelevel.keys())

        result = sum(nodelevel[maxLevel])

        return result