[BOJ] 백준 2623 음악프로그램

문제 링크

www.acmicpc.net/problem/2623

2623번: 음악프로그램

 

풀이

방향성 있는 그래프가 주어집니다. 출연 순서를 정할 수 없는 상황은 그래프에 사이클이 있는 경우입니다.

 

위상 정렬을 돌리되, 방문할 수 없는 정점이 있다는 말은 그 정점의 indegree가 0이 되지 않는다는 말이고, 사이클이 있다는 말과 동일합니다.

 

위상 정렬 후 모든 정점을 방문할 수 있는지 확인하여 답을 출력해줍시다.

 

정답 코드

#include <bits/stdc++.h>

using namespace std;
using vi = vector<int>;
using vvi = vector<vi>;

int n, m;
int indeg[1111];
vvi p;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    cin >> n >> m;
    p.resize(n + 1);
    while (m--) {
        int mn;
        cin >> mn;
        int prev = 0, now;
        while (mn--) {
            cin >> now;
            if (prev) {
                p[prev].push_back(now);
                indeg[now]++;
            }
            prev = now;
        }
    }
    queue<int> q;
    for (int i = 1; i < n + 1; i++) {
        if (indeg[i] == 0) {
            q.push(i);
        }
    }
    vi ans;
    while (!q.empty()) {
        int now = q.front();
        q.pop();
        ans.push_back(now);
        for (int next: p[now]) {
            if (--indeg[next] == 0) {
                q.push(next);
            }
        }
    }
    if (ans.size() != n) {
        cout << 0;
        return 0;
    }
    for (int i: ans) {
        cout << i << '\n';
    }
}