Today's Challenge

https://leetcode.com/problems/binary-tree-level-order-traversal/

Binary Tree Level Order Traversal - LeetCode

이것도 며칠 전에 푼 전위 순회를 활용한 문제.
머릿속에서는 그래도 어떻게 풀면 되겠지 했는데.. 손이 안 가서 솔루션 참고 ㅠㅠ

class Solution:
    def levelOrder(self, root: Optional[TreeNode]) -> List[List[int]]: 
        
        def trans(k):
            if len(k) == 0:
                return            
            ans, nk = [], []
            for node in k:
                ans.append(node.val)
                
                if node.left:
                    nk.append(node.left)
                if node.right:
                    nk.append(node.right)            
            res.append(ans)
            trans(nk)
            
        res = []
        if root:
            trans([root])
        return res