11725. 트리의 부모 찾기 (Java)

문제가 이해 되지 않아 꽤 힘들었던 문제

말그대로 2번 노드부터 N번 노드까지의 부모를 구하라는 문제였다.

DFS를 사용하면 금방 쉽게 풀수 있으나 System.out 보다 StringBuilder를 사용하는것을 추천한다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.StringTokenizer;

public class Main {

    static int N;
    static ArrayList<Integer>[] list;
    static int[] answer;
    static boolean[] isVisited;

    static void DFS(int s){
        for (int e:list[s]){
            if(isVisited[e]) continue;
            else {
                isVisited[e] = true;
                answer[e] = s;
                DFS(e);
            }
        }
    }

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        N = Integer.parseInt(br.readLine());
        answer = new int[N+1];
        isVisited= new boolean[N+1];
        isVisited[1]= true;
        list = new ArrayList[N+1];

        for(int i = 1 ; i <= N ; i++){
            list[i]=new ArrayList<>();
        }
        for(int i = 0; i < N-1; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());

            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());

            list[a].add(b);
            list[b].add(a);
        }
        DFS(1);
        for(int i = 2 ; i <= N ; i++){
            sb.append(answer[i]+"\n");
        }
        System.out.println(sb);
    }
}