1068-트리

https://www.acmicpc.net/problem/1068

1068번: 트리

와... 이문제는 트리를 어떻게 만들어야 하는지 몰라서 시간을 많이 잡아먹은 문제였다.

나는 트리를 2차원 배열을 사용하여 각각 해당하는 노드의 위치를 넣어준 후, DFS 알고리즘을 사용하여 노드를 탐방하면서 지워야 할 숫자가 나오면 그 부분은 스택에 추가시키지 않는 방향으로 코드를 만들어보았다.

package Making;


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


public class N1068B {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        Stack<Integer> stack = new Stack<>();
        ArrayList<ArrayList<Integer>> Node = new ArrayList<>();
        int root = 0;
        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            Node.add(new ArrayList<>(i));
        }
        for (int i = 0; i < N; i++) {
            int a = Integer.parseInt(st.nextToken());
            if(a == -1){
                root = i;
                continue;
            }
            Node.get(a).add(i);
        }
        int target = Integer.parseInt(br.readLine());
        int ans = 0;
        if(root == target){
            System.out.println(0);
            return;
        }
        stack.add(root);
        while(!stack.isEmpty()){
            boolean nothing = true;
            int cur = stack.pop();
            for (int i = 0; i < Node.get(cur).size(); i++) {
                if(Node.get(cur).get(i) == target){
                    continue;
                }
                stack.add(Node.get(cur).get(i));
                nothing = false;
            }
            if(nothing){
                ans++;
            }
        }
        System.out.println(ans);
    }
}

코드는 길지 않지만 생각하는데 시간을 많이 쓴 문제였다.