취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 2623번 문제인 음악프로그램이다.
문제는 아래 링크를 확인하자.

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

주어지는 출연순서 관계들을 하나의 그래프로 나타내고, 위상정렬을 시도한다.

만약 순서를 정할 수 있다면(DAG라면) 그 결과로 "가능한 출연순서" 중 하나를 얻을 것이다.

그렇지 않다면, 그래프에 사이클이 포함되어있으므로 얻은 순서의 크기가 N보다 작게 되고, 이를 확인하여 0을 출력한다.

아래는 제출한 소스코드이다.

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

vector<int> G[1001];
int cnt[1001];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	int N, M; cin >> N >> M;
	while (M--) {
		int K; cin >> K;
		if (K == 0) continue;
		K--;
		int old; cin >> old;
		while (K--) {
			int current; cin >> current;
			G[old].push_back(current);
			cnt[current]++;
			old = current;
		}
	}

	queue<int> que;
	queue<int> ans;

	for (int i = 1; i <= N; i++) {
		if (!cnt[i]) que.push(i);
	}

	while (!que.empty()) {
		int current = que.front(); que.pop();
		ans.push(current);

		for (auto node : G[current]) {
			cnt[node]--;
			if (!cnt[node]) que.push(node);
		}
	}

	if (ans.size() < N) cout << 0;
	else {
		while (!ans.empty()) {
			cout << ans.front() << '\n';
			ans.pop();
		}
	}
}