[백준] 1005번 - ACM Craft [Java]

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

 

[백준] 1005번 - ACM Craft [Java]

[백준] 1005번 - ACM Craft [Java]

[백준] 1005번 - ACM Craft [Java]

[백준] 1005번 - ACM Craft [Java]

[백준] 1005번 - ACM Craft [Java]

---

1.  아이디어

 

위상 정렬과 건설 시간을 저장하는 다이나믹 프로그래밍을 활용하면 해결할 수 있다.

---

2. 문제풀이

 

그림에서 위상 정렬의 힌트를 얻을 수 있어서 위상 정렬로 접근했다. 각 건물의 건설 시간은 dist 배열에 저장했는데 dist 배열은 선행 건물이 없는 건물은 해당 건물의 건설시간으로 초기화하면 되고 선행 건물이 있는 경우 (선행 건물을 건설하는데 걸리는 총 시간 + 해당 건물을 건설하는데 걸리는 시간)과 현재까지 계산된 해당 건물을 건설하는데 걸리는 총 시간 중 큰 값이 해당 건물을 건설하는데 걸리는 총 시간이 된다.

---

3. 코드

 

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st;

        int T = Integer.parseInt(br.readLine());

        for (int tc = 1; tc <= T; tc++) {
            st = new StringTokenizer(br.readLine());
            int N = Integer.parseInt(st.nextToken());
            int K = Integer.parseInt(st.nextToken());

            int[] D = new int[1 + N];
            st = new StringTokenizer(br.readLine());
            for (int i = 1; i <= N; i++) {
                D[i] = Integer.parseInt(st.nextToken());
            }

            List<List<Integer>> adj = new ArrayList<>();
            for (int i = 0; i <= N; i++) {
                adj.add(new ArrayList<>());
            }

            int[] indegree = new int[1 + N];

            for (int i = 0; i < K; i++) {
                st = new StringTokenizer(br.readLine());
                int X = Integer.parseInt(st.nextToken());
                int Y = Integer.parseInt(st.nextToken());

                adj.get(X).add(Y);
                indegree[Y]++;
            }

            int W = Integer.parseInt(br.readLine());

            sb.append(topologicalSort(W, D, N, adj, indegree)).append("\n");
        }

        bw.write(sb.toString());
        bw.flush();
    }

    private static int topologicalSort(int target, int[] D, int N, List<List<Integer>> adj, int[] indegree) {
        int[] dist = new int[1 + N];

        Queue<Integer> q = new ArrayDeque<>();

        for (int i = 1; i <= N; i++) {
            if (indegree[i] == 0) {
                q.add(i);
                dist[i] = D[i];
            }
        }

        while (!q.isEmpty()) {
            int node = q.remove();

            for (int next : adj.get(node)) {
                indegree[next]--;
                if (indegree[next] == 0) q.add(next);

                dist[next] = Math.max(dist[next], dist[node] + D[next]);
            }
        }

        return dist[target];
    }

}

---

4. 후기