[LeetCode] 102. Binary Tree Level Order Traversal, Medium

1. 문제

• 이진 트리가 주어질 때, level 순회한 결과를 level 별로 묶어서 반환하라.

2. 해결

/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function levelOrder(root: TreeNode | null): number[][] {
    if(!root) return []
    const queue: TreeNode[] = [root];

    const result: number[][] = [];

    while(queue.length > 0) {
        const levelSize = queue.length;
        const level = []
        for(let i = 0; i< levelSize; i++) {
            const node = queue.shift();
            level.push(node.val);

            if(node.left) queue.push(node.left);
            if(node.right) queue.push(node.right);
        }

        result.push(level);
    }
    
    return result;
};

• 단순 level 순회는 간단하게 구현했지만, level 별로 노드를 묶어서 반환하는 부분을 고생했다.
• 큐에 담긴 level 노드 개수만큼 반복하면서 담은 배열을 result 배열에 담아주면 된다.
• 3 -> 9, 20 -> 15, 7 순으로 Queue에 들어가면서 반복되게 된다.