티스토리 뷰
728x90
문제 링크
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
'알고리즘 > 문제 풀이' 카테고리의 다른 글
[BOJ] 백준 7453 합이 0인 네 정수 (0) | 2021.03.08 |
---|---|
[BOJ] 백준 2887 행성 터널 (0) | 2021.03.08 |
[BOJ] 백준 2098 외판원 순회 (0) | 2021.03.08 |
[BOJ] 백준 2342 Dance Dance Revolution (0) | 2021.03.08 |
[BOJ] 백준 2252 줄 세우기 (0) | 2021.03.08 |