티스토리 뷰

728x90

문제 링크

www.acmicpc.net/problem/2623

 

2623번: 음악프로그램

첫째 줄에는 가수의 수 N과 보조 PD의 수 M이 주어진다. 가수는 번호 1, 2,…,N 으로 표시한다. 둘째 줄부터 각 보조 PD가 정한 순서들이 한 줄에 하나씩 나온다. 각 줄의 맨 앞에는 보조 PD가 담당한

www.acmicpc.net

 

풀이

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

 

위상 정렬을 돌리되, 방문할 수 없는 정점이 있다는 말은 그 정점의 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';
}
}
728x90
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
«   2025/03   »
1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31
글 보관함