[BOJ] 2623 음악프로그램 - JAVA

1. 문제

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들이 만든 순서들이 입력으로 주어질 때, 전체 가수의 순서를 정하는 프로그램을 작성하시오.

---

2. 풀이

입력

count[i]는 i번째 가수 직전에 노래를 불러야 하는 가수의 수다.

order[i][j]는 i번째 가수 다음에 j번째 가수가 노래를 불러야 한다는 것을 의미한다.

PD들의 가장 첫번째 가수이거나 이미 직전가수→현재 가수 경우가 있는 게 아니라면 count[now]+1하고 order[prev][now]도 true로 바꾼다.

BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(in.readLine());
int N = Integer.parseInt(st.nextToken());
int M = Integer.parseInt(st.nextToken());

boolean[][] order = new boolean[N+1][N+1];
int[] count = new int[N+1];

int prev, now, m;
for(int i=0;i<M;i++){
   st = new StringTokenizer(in.readLine());
   m = Integer.parseInt(st.nextToken());
   prev = 0;
   for(int j=0;j<m;j++){
      now = Integer.parseInt(st.nextToken());
      if(prev != 0 && !order[prev][now]) {
         count[now]++;
         order[prev][now] = true;
      }
      prev = now;
   }
}

 

탐색

1.  count[i]가 0인 가수(=먼저 불러야 하는 가수가 없는 가수)를 queue에 추가한다.

2.  큐에서 뽑은 요소를 answer에 더한다.

3.  현재 가수 다음에 불러야 하는 가수들의 count를 하나씩 줄이면서, 줄인 결과가 0이라면 queue에 더한다.

Queue<Integer> queue = new ArrayDeque<>();
for(int i=1;i<=N;i++) {
   if(count[i]==0) queue.offer(i);
}

ArrayList<Integer> answer = new ArrayList<>();
while(!queue.isEmpty()){
   now = queue.poll();
   answer.add(now);

   for(int i=1;i<=N;i++){
      if(order[now][i]){
         if(--count[i]==0) queue.offer(i);
      }
   }
}

 

출력

answer의 길이는 항상 N이어야 한다. 만약 N이 안됐다면 사이클이 발생한 것이므로 0을 출력하고 끝낸다.

정상적인 경우 ans 요소를 하나씩 sb에 더해 출력한다.

if(answer.size()!=N) {
   System.out.println(0);
   return;
}

StringBuilder sb = new StringBuilder();
for(int ans:answer) sb.append(ans).append('\n');
System.out.print(sb);

 

전체 코드

import java.io.*;
import java.util.*;

public class BOJ_2623_음악프로그램 {
   public static void main(String[] args) throws IOException {
      BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
      StringTokenizer st = new StringTokenizer(in.readLine());
      int N = Integer.parseInt(st.nextToken());
      int M = Integer.parseInt(st.nextToken());

      boolean[][] order = new boolean[N+1][N+1];
      int[] count = new int[N+1];

      int prev, now, m;
      for(int i=0;i<M;i++){
         st = new StringTokenizer(in.readLine());
         m = Integer.parseInt(st.nextToken());
         prev = 0;
         for(int j=0;j<m;j++){
            now = Integer.parseInt(st.nextToken());
            if(prev != 0 && !order[prev][now]) {
               count[now]++;
               order[prev][now] = true;
            }
            prev = now;
         }
      }

      Queue<Integer> queue = new ArrayDeque<>();
      for(int i=1;i<=N;i++) {
         if(count[i]==0) queue.offer(i);
      }

      ArrayList<Integer> answer = new ArrayList<>();
      while(!queue.isEmpty()){
         now = queue.poll();
         answer.add(now);

         for(int i=1;i<=N;i++){
            if(order[now][i]){
               if(--count[i]==0) queue.offer(i);
            }
         }
      }

      if(answer.size()!=N) {
         System.out.println(0);
         return;
      }

      StringBuilder sb = new StringBuilder();
      for(int ans:answer) sb.append(ans).append('\n');
      System.out.print(sb);
   }
}

---

3. 결과

PriorityQueue를 쓰는 과정에서 약간 꼬였다.