[백준] 2252번 - 줄 세우기 (파이썬)

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

# 줄 세우기
from collections import deque
import sys
input = sys.stdin.readline
N, M = map(int, input().split())
arr = [[] for _ in range(N+1)]
front = [0] * (N+1)
for _ in range(M):
    a, b = map(int, input().split())
    arr[a].append(b)
    front[b] += 1


def sort():
    result = []
    q = deque()
    for k in range(1, len(front)):
        if front[k] == 0:
            q.append(k)
    while q:
        now = q.popleft()
        result.append(now)
        for nnow in arr[now]:
            front[nnow] -= 1
            if front[nnow] == 0:
                q.append(nnow)
    print(*result)


sort()

 

위상정렬

1. 입력을 받으면서 인접 리스트를 만들고, 앞에 몇 명이 있는지 front리스트에 기록

2. q에 앞에 있어야 할 사람이 0인 사람들을 append하고, q가 빌 때까지

  - 인접 리스트에 연결된 사람의 front에서 1을 빼주고 0이면 q에 append

3. 출력

 

선수과목이 있는 강의 순서 등, 앞에 꼭 어떠한 것이 있을 때의 정렬 : 위상정렬