백준, 2252, 줄세우기, 위상정렬

Problem2252.java

package sdk.backjun.graph;

import java.util.HashMap;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

/**
 * 백준, 2252, 위상정렬, 줄세우기
 * 
 * @author whitebeardk
 *
 */
public class Problem2252 {

	static int N; // 문제의 수
	static int M; // 정보의 개수
	static HashMap<Integer, LinkedList<Integer>> maps = new HashMap<>();
	static int[] in;

	public static void main(String[] args) {

		Scanner sc = new Scanner(System.in);
		N = sc.nextInt();
		M = sc.nextInt();
		in = new int[N + 1];

		for (int i = 0; i < M; i++) {
			int from = sc.nextInt();
			int to = sc.nextInt();

			if (!maps.containsKey(from)) {
				LinkedList<Integer> list = new LinkedList<>();
				maps.put(from, list);
			}

			// 노드의 연결상태를 링크드 리스트로 표현 
			maps.get(from).add(to);
			// 입력으로 들어오는 간선의 상태를 기록 
			in[to]++;
		}
		sc.close();

		Queue<Integer> queue = new LinkedList<>();

		// 초기 시작점을 기록 
		for (int i = 1; i < N + 1; i++) {
			if (in[i] == 0)
				queue.offer(i);
		}

		while (true) {
			if (queue.size() == 0)
				break;

			int from = queue.poll();

			System.out.print(from + " ");

			// 노드가 간선을 가지고 있으면 큐에 추가 
			if (maps.containsKey(from)) {
				for (int to : maps.get(from)) {

					if (--in[to] == 0)
						queue.add(to);

				}
			}
		}

	}
}