Leetcode Java N-ary Tree Preorder Traversal

문제

Link

코드

class Solution {

  public List<Integer> preorder(Node root) {
    List<Integer> result = new ArrayList<>();
    this.dfs(root, result);
    return result;
  }

  private void dfs(Node root, List<Integer> list) {
    if (root != null) {
      list.add(root.val);
      if (root.children != null) {
        for (Node node : root.children) {
          this.dfs(node, list);
        }
      }
    }
  }

}

결과

Link

설명

1. root를 preorder 순으로 val 값들을 반환하는 문제이다.

2. result는 preorder 순으로 val 값을 넣을 변수로, ArrayList로 초기화한다.

3. 4번에서 정의한 dfs(Node root, List list)에 result를 넣어 수행한다.

4. root를 preorder 순으로 val 값을 넣기 위한 dfs(Node root, List list)를 정의한다
   • root가 null이 아닌 경우 아래를 수행한다.
   • list의 root의 val 값을 먼저 넣어준다.
   • root의 children이 null이 아닌 경우, children을 순차적으로 root 자리에 넣어 재귀 호출을 한다.
5. root를 preorder 순으로 val 값들을 넣은 result를 주어진 문제의 결과로 반환한다.

소스

Sample Code는 여기에서 확인 가능합니다.