Today's Challenge

https://leetcode.com/problems/average-of-levels-in-binary-tree/

Average of Levels in Binary Tree - LeetCode

Queue를 이용한 BFS로 간단히 풀 수 있음.

class Solution:
    def averageOfLevels(self, root: Optional[TreeNode]) -> List[float]:
        qu = []
        res = []
        qu.append(root)
        
        while qu:
            l = len(qu)
            temp = 0
            for i in range(l):
                node = qu.pop(0)
                temp += node.val
                if node.left:
                    qu.append(node.left)
                if node.right:
                    qu.append(node.right)
            res.append(temp/l)
            
        return res