BOJ)1766 문제집

문제: icpc.me/1766

문제집을 푸는 순서가 주어지는데 이때 가능하면 쉬운 문제부터 풀어야 할 경우 문제를 푸는 순서를 출력하는 문제이다.

우리는 문제집을 푸는 순서가 주어질 때 topological sort를 통하여 순서를 출력할 수 있다.

하지만 가능하면 쉬운 문제를 풀어야 하니 topology가 여러개일 경우 작은 순서를 먼저 출력해줘야 한다. 이는 min heap을 통하여 해결 가능하다.

#include <cstdio>
#include <algorithm>
#include <vector>
#include <queue>
#define MAX_N 32000
using namespace std;
int n, m, a, b, in[MAX_N + 1];
vector<vector<int>> vt;
priority_queue<int> pq;
int main() {
    scanf("%d%d", &n, &m);
    vt.resize(n + 1);
    for (int i = 0; i < m; i++) {
        scanf("%d%d", &a, &b);
        vt[a].push_back(b);
        in[b]++;
    }
    for (int i = 1; i <= n; i++) {
        if (!in[i])
            pq.push(-i);
    }
    while (pq.size()) {
        int here = -pq.top();
        pq.pop();
        printf("%d ", here);
        for (int there : vt[here]) {
            in[there]--;
            if (!in[there])
                pq.push(-there);
        }
    }
    return 0;
}