[LeetCode] 429번 문제를 풀어보았다. (ft. java)

2022년 09월 05일 월요일 - 비야 그만와라...

오늘 올려볼 문제는 429번 N-ary Tree Level Order Traversal 이라는 문제이다.

사진을 클릭하면 해당 문제로 이동합니다.

오늘도 LeetCode 사이트 오늘의 문제를 가지고 왔다.

태풍이 오는데 재택 근무 안해??

입력

사진에서도 볼 수 있듯이 Tree의 루트 노드가 입력으로 들어온다.

---

풀이 및 코드

각 레벨에 맞게끔 분류해서 리턴하는 문제다.

오늘은 처음부터 정답을 생각해냈다.

그냥 bfs 썼다.

이제 코드를 봐보자!

풀이코드

class Solution {
    public List<List<Integer>> levelOrder(Node root) {
        if(root == null) return new ArrayList<>();
        
        Queue<Node> q = new LinkedList<>();
        q.add(root);
        
        List<List<Integer>> result = new ArrayList<>();
        
        while(!q.isEmpty()) {
            int size = q.size();
            
            List<Integer> temp = new ArrayList<>();
            
            for(int i = 0; i < size; i++) {
                Node now = q.poll();
                
                temp.add(now.val);
                
                List<Node> list = now.children;
                
                for(int j = 0; j < list.size(); j++) {
                    q.add(list.get(j));
                }
            }
            
            result.add(temp);
        }
        
        return result;
    }
}

---

제출 화면

비가 이렇게 많이 오는데 회사를 가야한다니... 두렵다....

사실 두렵지는 않고 너무 귀찮다....

내일도 문제를 풀어서 블로그에 글을 쓸 수 있으면 좋겠다.