Algorithm/BOJ
[백준] 2252번 - 줄 세우기 (파이썬)
_temp
2022. 1. 24. 12:37
# 줄 세우기
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. 출력
선수과목이 있는 강의 순서 등, 앞에 꼭 어떠한 것이 있을 때의 정렬 : 위상정렬