[백준] 2056번 - 작업 [Java]

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

 

[백준] 2056번 - 작업 [Java]

[백준] 2056번 - 작업 [Java]

---

1.  아이디어

 

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

---

2. 문제풀이

 

위상 정렬 과정에서 선행 작업이 없는 경우 해당 작업 시간을 선행 작업이 있는 경우 선행 작업을 하는데 걸리는 총 시간과 해당 작업을 하는데 걸리는 시간의 합과 현재까지 구한 해당 작업을 하는데 걸리는 총 시간 중 더 큰 값이 해당 작업을 하는데 걸리는 총 시간이 된다. 이를 통해 모든 작업이 각각 완료되는데 걸리는 총 시간을 배열로 구할 수 있었고 이 중 가장 큰 값이 모든 작업을 완료하는데 걸리는 총 시간이 됐다. N번 작업보다 번호가 작은 작업이 더 오래걸릴 수 있음에 주의해야한다.

---

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));
        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());

            int len = Integer.parseInt(st.nextToken());
            while (st.hasMoreTokens()) {
                int token = Integer.parseInt(st.nextToken());

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

        int ans = topologicalSort(N, adj, indegree, weight);
        System.out.println(ans);
    }

    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]);
            }
        }

        int max = Integer.MIN_VALUE;
        for (int i = 1; i <= N; i++) {
            max = Math.max(max, dist[i]);
        }
        return max;
    }

}

---

4. 후기