[백준] 1766번 - 문제집 [Java]

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

 

[백준] 1766번 - 문제집 [Java]

[백준] 1766번 - 문제집 [Java]

---

1.  아이디어

 

위상 정렬과 우선순위 큐를 활용하면 해결할 수 있다.

---

2. 문제풀이

 

위상 정렬만으로 조건 1과 조건 2는 만족할 수 있는데 조건 3이 안될 수가 있다. 위상 정렬의 Queue에 있는 원소들은 모두 동일한 단계이므로 Queue를 우선순위 큐로 바꾼 위상정렬을 적용하면 조건 3도 만족시킬 수 있었다.

---

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 = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int M = 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 < M; i++) {
            st = new StringTokenizer(br.readLine());
            int A = Integer.parseInt(st.nextToken());
            int B = Integer.parseInt(st.nextToken());

            adj.get(A).add(B);
            indegree[B]++;
        }

        PriorityQueue<Integer> pq = new PriorityQueue<>();

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

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

            sb.append(node).append(" ");

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

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

---

4. 후기