[Baekjoon Online Judge] 백준 2252번 줄 세우기(Python)

© 2020 All Rights Reserved. 주식회사 스타트링크

[Baekjoon Online Judge] 백준 2252번 줄 세우기

(Python)

(글쓴날 : 2020.06.16)

 

---

* Baekjoon Online Judge, 백준 2252번 문제 Python 언어 풀이입니다.

* 소스 코드의 저작권은 글쓴이에게 있습니다.

---

 

 

백준 2252번 줄 세우기

---

1) 문제

문제 링크 : https://www.acmicpc.net/problem/2252

2252번: 줄 세우기

---

2) 풀이 과정

* 시간 복잡도 : O(V+E)

 

N명의 학생들 중 일부 학생들의 키를 비교한 순서가 주어질 때, 가능한 전체 학생들의 키 순서를 구하는 문제입니다.

 

저의 경우, 위상 정렬을 적용하였고, Python을 사용했습니다.

주어지는 일부 학생들의 키 순서를 인접 리스트에 구현한 뒤, 위상 정렬을 적용하여 가능한 전체 학생들의 키 순서 중 하나를 구해 문제를 해결했습니다.

---

3) 코드

 

* Python 코드

from collections import deque
import sys
input = sys.stdin.readline
 
 
def topologicalSort():
    for _ in range(N):
        if not dq:
            return
        
        target = dq.popleft()
        ans.append(target)
 
        for v in adjList[target]:
            indegree[v] -= 1
 
            if not indegree[v]:
                dq.append(v)
                
 
N, M = map(int, input().split())
 
ans = []
indegree = [0] * (N + 1)
adjList = [[] for _ in range(N + 1)]
 
for _ in range(M):
    A, B = map(int, input().split())
 
    adjList[A].append(B)
    indegree[B] += 1
 
dq = deque()
for i in range(1, N + 1):
    if not indegree[i]:
        dq.append(i)
 
topologicalSort()
 
print(*ans)

---