[백준] 2252번 - 줄 세우기 [Java]

https://www.acmicpc.net/problem/2252

 

[백준] 2252번 - 줄 세우기 [Java]

[백준] 2252번 - 줄 세우기 [Java]

---

1.  아이디어

 

위상 정렬을 활용하면 간단하게 해결할 수 있다.

---

2. 문제풀이

 

키가 작은 친구에서 키가 큰 친구로 간선이 향하는 방향 비순환 그래프로 생각할 수 있으므로 위상 정렬을 구현했고 노드를 꺼내는 과정에서 StringBuilder에 담은 후 출력하는 방식으로 구현했다.

---

3. 코드

 

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        List<List<Integer>> adj = new ArrayList<>();
        for (int i = 0; i <= N; i++) {
            adj.add(new ArrayList<>());
        }

        int[] indegree = new int[1 + N];

        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int A = Integer.parseInt(st.nextToken());
            int B = Integer.parseInt(st.nextToken());

            adj.get(A).add(B);
            indegree[B]++;
        }

        Queue<Integer> q = new ArrayDeque<>();

        for (int i = 1; i <= N; i++) {
            if (indegree[i] == 0) q.add(i);
        }

        while (!q.isEmpty()) {
            int node = q.remove();

            sb.append(node).append(" ");

            for (int next : adj.get(node)) {
                indegree[next]--;
                if (indegree[next] == 0) q.add(next);
            }
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기