[백준] 1516번 - 게임 개발 [Java]

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

 

[백준] 1516번 - 게임 개발 [Java]

[백준] 1516번 - 게임 개발 [Java]

---

1.  아이디어

 

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

---

2. 문제풀이

 

이전 ACM Craft 문제와 유사한 문제로 동일하게 접근했다.([코딩테스트 준비/백준] - [백준] 1005번 - ACM Craft [Java]) 각 건물의 건설 시간은 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 N = Integer.parseInt(br.readLine());

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

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

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

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

            while (st.hasMoreTokens()) {
                int token = Integer.parseInt(st.nextToken());
                if (token == -1) break;

                adj.get(token).add(i);
                indegree[i]++;
            }
        }

        int[] dist = topologicalSort(N, adj, indegree, weight);
        for (int i = 1; i <= N; i++) {
            sb.append(dist[i]).append("\n");
        }

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

    private static int[] topologicalSort(int N, List<List<Integer>> adj, int[] indegree, int[] weight) {
        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] = weight[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] + weight[next]);
            }
        }

        return dist;
    }

}

---

4. 후기