-
Notifications
You must be signed in to change notification settings - Fork 848
/
Copy path5.py
79 lines (71 loc) · 2.65 KB
/
5.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
from collections import deque
# 테스트 케이스(Test Case)만큼 반복
for tc in range(int(input())):
# 노드의 개수 입력 받기
n = int(input())
# 모든 노드에 대한 진입차수는 0으로 초기화
indegree = [0] * (n + 1)
# 각 노드에 연결된 간선 정보를 담기 위한 인접 행렬 초기화
graph = [[False] * (n + 1) for i in range(n + 1)]
# 작년 순위 정보 입력
data = list(map(int, input().split()))
# 방향 그래프의 간선 정보 초기화
for i in range(n):
for j in range(i + 1, n):
graph[data[i]][data[j]] = True
indegree[data[j]] += 1
# 올해 변경된 순위 정보 입력
m = int(input())
for i in range(m):
a, b = map(int, input().split())
# 간선의 방향 뒤집기
if graph[a][b]:
graph[a][b] = False
graph[b][a] = True
indegree[a] += 1
indegree[b] -= 1
else:
graph[a][b] = True
graph[b][a] = False
indegree[a] -= 1
indegree[b] += 1
# 위상 정렬(Topology Sort) 시작
result = [] # 알고리즘 수행 결과를 담을 리스트
q = deque() # 큐 기능을 위한 deque 라이브러리 사용
# 처음 시작할 때는 진입차수가 0인 노드를 큐에 삽입
for i in range(1, n + 1):
if indegree[i] == 0:
q.append(i)
certain = True # 위상 정렬 결과가 오직 하나인지의 여부
cycle = False # 그래프 내 사이클이 존재하는지 여부
# 정확히 노드의 개수만큼 반복
for i in range(n):
# 큐가 비어 있다면 사이클이 발생했다는 의미
if len(q) == 0:
cycle = True
break
# 큐의 원소가 2개 이상이라면 가능한 정렬 결과가 여러 개라는 의미
if len(q) >= 2:
certain = False
break
# 큐에서 원소 꺼내기
now = q.popleft()
result.append(now)
# 해당 원소와 연결된 노드들의 진입차수에서 1 빼기
for j in range(1, n + 1):
if graph[now][j]:
indegree[j] -= 1
# 새롭게 진입차수가 0이 되는 노드를 큐에 삽입
if indegree[j] == 0:
q.append(j)
# 사이클이 발생하는 경우(일관성이 없는 경우)
if cycle:
print("IMPOSSIBLE")
# 위상 정렬 결과가 여러 개인 경우
elif not certain:
print("?")
# 위상 정렬을 수행한 결과 출력
else:
for i in result:
print(i, end=' ')
print()