[leetcode] 429. N-ary Tree Level Order Traversal

leetcode.com/problems/n-ary-tree-level-order-traversal/

N-ary Tree Level Order Traversal - LeetCode

이 문제는 bfs의 기본적인 문제입니다.

queue에 시작점을 넣어두고 해당 시작점과 연결된 지점들을 다시 넣어주고를 반복합니다.

반복할 때마다 1depth씩 내려갑니다.

class Solution {
    public List<List<Integer>> levelOrder(Node root) {
        List<List<Integer>> result = new ArrayList<>();
        if(root == null) {
            return result;
        }
        
        Queue<Node> queue = new LinkedList<>();
        queue.offer(root);

        while(!queue.isEmpty()) {
            List<Integer> list = new ArrayList<>();
            int size = queue.size();
            for(int i=0; i<size; i++) {
                Node node = queue.poll();
                for(Node child : node.children) {
                    queue.offer(child);
                }
                list.add(node.val);
            }
            if(list.size() > 0) {
                result.add(list);
            }
        }
        return result;
    }
}