🎲문제
주소:백준 2623
인터넷 방송 KOI(Korea Open Internet)의 음악 프로그램 PD인 남일이는 자기가 맡은 프로그램 '뮤직 KOI'에서 가수의 출연 순서를 정하는 일을 매우 골치 아파한다. 순서를 정하기 위해서는 많은 조건을 따져야 한다.
그래서 오늘 출연 예정인 여섯 팀의 가수에 대해서 남일이가 보조 PD 세 명에게 각자 담당한 가수의 출연 순서를 정해오게 하였다. 보조 PD들이 가져온 것은 아래와 같다.
- 1 4 3
- 6 2 5 4
- 2 3
첫 번째 보조 PD는 1번 가수가 먼저, 다음에 4번 가수, 다음에 3번 가수가 출연하기로 순서를 정했다. 두 번째 보조 PD는 6번, 2번, 5번, 4번 순으로 자기 담당 가수들의 순서를 정했다. 한 가수를 여러 보조 PD가 담당할 수도 있다. 마지막으로, 세 번째 보조 PD는 2번 먼저, 다음에 3번으로 정했다.
남일이가 할 일은 이 순서들을 모아서 전체 가수의 순서를 정하는 것이다. 남일이는 잠시 생각을 하더니 6 2 1 5 4 3으로 순서를 정했다. 이렇게 가수 순서를 정하면 세 보조 PD가 정해온 순서를 모두 만족한다. 물론, 1 6 2 5 4 3으로 전체 순서를 정해도 괜찮다.
경우에 따라서 남일이가 모두를 만족하는 순서를 정하는 것이 불가능할 수도 있다. 예를 들어, 세 번째 보조 PD가 순서를 2 3 대신에 3 2로 정해오면 남일이가 전체 순서를 정하는 것이 불가능하다. 이번에 남일이는 우리 나라의 월드컵 4강 진출 기념 음악제의 PD를 맡게 되었는데, 출연 가수가 아주 많다. 이제 여러분이 해야 할 일은 보조 PD들이 가져 온 순서들을 보고 남일이가 가수 출연 순서를 정할 수 있도록 도와 주는 일이다.
보조 PD들이 만든 순서들이 입력으로 주어질 때, 전체 가수의 순서를 정하는 프로그램을 작성하시오.
⌨ 입력
첫째 줄에는 가수의 수 N과 보조 PD의 수 M이 주어진다. 가수는 번호 1, 2,…,N 으로 표시한다. 둘째 줄부터 각 보조 PD가 정한 순서들이 한 줄에 하나씩 나온다. 각 줄의 맨 앞에는 보조 PD가 담당한 가수의 수가 나오고, 그 뒤로는 그 가수들의 순서가 나온다. N은 1이상 1,000이하의 정수이고, M은 1이상 100이하의 정수이다.
✅ 풀이
풀이방법
간단한 위상정렬을 통해 해결하였다. 부모 자식 관계를 벡터로 저장해 놓고 큐에 넣어 부모 관계를 하나씩 지워가며 답을 저장하고, 사이클이 하나라도 존재한다면 전체 답의 크기가 N보다 작아지기 때문에 결과 답의 크기가 N보다 작다면 0을 출력하는 방식으로 해결했다.
-
N과 M의 값을 입력받는다.
-
가수의 수를 입력받고 for문으로 숫자 하나씩 입력받는다.
-
첫번째 수를 임시로 저장하고, 두 번째 수를 입력받을 때 parent[두번째 수]에 첫 번째 수를 넣고 v[첫 번째 수]에 두 번째 수를 넣는다.
-
두번째 수를 다시 임시로 저장하고 세번째 수를 입력받을 때 parent[세번째 수]에 두 번째 수를 넣고 v[두 번째 수]에 세 번째 수를 넣는다.
-
위를 반복하여 모든 순서를 입력받는다.
void init() {
cin >> N >> M;
for (int i = 0; i < M; i++) {
int num, tmp;
cin >> num;
cin >> tmp;
for (int j = 0; j < num-1; j++) {
int x;
cin >> x;
parent[x].push_back(tmp);
v[tmp].push_back(x);
tmp = x;
}
}
}- 큐에 parent[0] - parent[n]까지 중 부모가 없는 요소를 큐에 집어넣는다.
- while(!queue.empty)를 사용하여 반복문을 돌린다.
- 가장 앞에있는 요소를 꺼내어(pop), 결과 벡터에 저장한다.
- v[cur][i](parent 반대 개념)을 확인하여 parent[cur의 자식]에 해당 요소가 있다면 지운다.
- 지웠을 때, 사이즈가 0이라면 큐에 집어넣는다.
- 반복한다.
void solve() {
for (int i = 1; i <= N; i++)
if (parent[i].size() == 0) q.push(i);
while (!q.empty()) {
int cur = q.front();
q.pop();
result.push_back(cur);
for (int i = 0; i < v[cur].size(); i++) {
int v_cur = v[cur][i];
for (int j = 0; j < parent[v_cur].size(); j++) {
if (parent[v_cur][j] == cur) {
parent[v_cur].erase(parent[v_cur].begin() + j);
if (parent[v_cur].size() == 0) q.push(v_cur);
break;
}
}
}
}
if (chk_available()) {
for (int i = 0; i < N; i++) {
cout << result[i] << "\n";
}
}
}
- 배열 전체 사이즈가 N이아니라면 0 출력, N이라면 순서대로 벡터에서 뽑아 출력
bool chk_available() {
if (result.size() != N) {
cout << "0";
return false;
}
return true;
}
전체 코드
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
int N, M;
queue<int> q;
vector<int> v[1001];
vector<int> parent[1001];
vector<int> result;
void init();
void solve();
bool chk_available();
void init() {
cin >> N >> M;
for (int i = 0; i < M; i++) {
int num, tmp;
cin >> num;
cin >> tmp;
for (int j = 0; j < num-1; j++) {
int x;
cin >> x;
parent[x].push_back(tmp);
v[tmp].push_back(x);
tmp = x;
}
}
}
void solve() {
for (int i = 1; i <= N; i++)
if (parent[i].size() == 0) q.push(i);
while (!q.empty()) {
int cur = q.front();
q.pop();
result.push_back(cur);
for (int i = 0; i < v[cur].size(); i++) {
int v_cur = v[cur][i];
for (int j = 0; j < parent[v_cur].size(); j++) {
if (parent[v_cur][j] == cur) {
parent[v_cur].erase(parent[v_cur].begin() + j);
if (parent[v_cur].size() == 0) q.push(v_cur);
break;
}
}
}
}
if (chk_available()) {
for (int i = 0; i < N; i++) {
cout << result[i] << "\n";
}
}
}
bool chk_available() {
if (result.size() != N) {
cout << "0";
return false;
}
return true;
}
int main() {
init();
solve();
return 0;
}