본문 바로가기
알고리즘/백준

[Java] BOJ 2623번 음악프로그램

by 컴공맨 2021. 4. 23.
 

2623번: 음악프로그램

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

www.acmicpc.net


풀이

위상정렬을 사용하여 해결했습니다.


코드

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

public class Main {
	public static int N, M;
	public static int[] in;
	public static int[][] adj;
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder("");
		
		StringTokenizer st = new StringTokenizer(br.readLine(), " ");
		
		N = Integer.parseInt(st.nextToken());
		M = Integer.parseInt(st.nextToken());
		
		in = new int[N];
		adj = new int[N][N];
		
		for (int i = 0; i < M; ++i) {
			st = new StringTokenizer(br.readLine(), " ");
			
			int num = Integer.parseInt(st.nextToken());
			int from = Integer.parseInt(st.nextToken()) - 1;
			
			for (int j = 1; j < num; ++j) {
				int to = Integer.parseInt(st.nextToken()) - 1;
				
				if (adj[from][to] == 0) {
					adj[from][to] = 1;
					in[to]++;					
				}
				
				from = to;
			}
		}
		
		Queue<Integer> q = new LinkedList<Integer>();
		for (int i = 0; i < N; ++i) {
			if (in[i] == 0) {
				q.offer(i);
			}
		}
		
		int cnt = 0;
		while (!q.isEmpty()) {
			// 큐를 돌다가 사이클발생이 가능!
			cnt++;
			
			int node = q.poll();
			sb.append(node + 1).append("\n");
			
			for (int i = 0; i < N; ++i) {
				if (adj[node][i] == 1) {
					in[i]--;
					
					if (in[i] == 0) {
						q.offer(i);
					}
				}
			}
		}
		
		System.out.println((cnt != N) ? 0 : sb);
	}
}

 

pyo7410/Algorithm

1일 1커밋을 목표로! Contribute to pyo7410/Algorithm development by creating an account on GitHub.

github.com

 

'알고리즘 > 백준' 카테고리의 다른 글

[Java] BOJ 9251번 LCS  (0) 2021.04.29
[Java] BOJ 1904번 01타일  (0) 2021.04.28
[Java] BOJ 11057번 오르막 수  (0) 2021.04.22
[Java] BOJ 9465번 스티커  (0) 2021.04.21
[Java] BOJ 1010번 다리 놓기  (0) 2021.04.16