명환 그래프 숙제 (#42)

Author: samkimuel

mhkim/10-2.py

@@ -0,0 +1,38 @@
+"""
+@file 10-2.py
+@brief 팀 결성
+@desc 서로소 집합 알고리즘
+"""
+
+# 팀 합치기
+def union_parent(parent, a, b):
+    a = find_parent(parent, a)
+    b = find_parent(parent, b)
+    if a < b:
+        parent[b] = a
+    else:
+        parent[a] = b
+
+
+# 같은 팀 여부 확인
+def find_parent(parent, x):
+    if parent[x] != x:
+        parent[x] = find_parent(parent, parent[x])
+    return parent[x]
+
+
+n, m = map(int, input().split())
+parent = [0] * (n + 1)
+
+for i in range(0, n + 1):
+    parent[i] = i
+
+for i in range(m):
+    op, a, b = map(int, input().split())
+    if op == 0:
+        union_parent(parent, a, b)
+    elif op == 1:
+        if find_parent(parent, a) == find_parent(parent, b):
+            print("YES")
+        else:
+            print("NO")

mhkim/10-3.py

@@ -0,0 +1,50 @@
+"""
+@file 10-3.py
+@brief 도시 분할 계획
+@desc 백준 1647번, 최소 신장 트리, 크루스칼 알고리즘
+
+전체 그래프에서 2개의 최소 신장 트리 만들기
+-> 크루스칼 알고리즘으로 최소 신장 트리 찾은 후 최소 신장 트리를 구성하는 간선 중에서 가장 비용이 큰 간선을 제거함
+"""
+
+
+def find_parent(parent, x):
+    if parent[x] != x:
+        parent[x] = find_parent(parent, parent[x])
+    return parent[x]
+
+
+def union_parent(parent, a, b):
+    a = find_parent(parent, a)
+    b = find_parent(parent, b)
+
+    if a < b:
+        parent[b] = a
+    else:
+        parent[a] = b
+
+
+v, e = map(int, input().split())
+parent = [0] * (v + 1)
+
+edges = []
+result = 0
+
+for i in range(1, v + 1):
+    parent[i] = i
+
+for i in range(e):
+    a, b, cost = map(int, input().split())
+    edges.append((cost, a, b))
+
+edges.sort()
+last = 0  # 최소 신장 트리 간선 중 가장 비용이 큰 간선
+
+for edge in edges:
+    cost, a, b = edge
+    if find_parent(parent, a) != find_parent(parent, b):
+        union_parent(parent, a, b)
+        result += cost
+        last = cost
+
+print(result - last)

mhkim/10-4.py

@@ -0,0 +1,46 @@
+"""
+@file 10-4.py
+@brief 커리큘럼
+@desc 위상 알고리즘
+
+위상 알고리즘을 수행, 더 오랜 시간 걸리는 경우의 시간 값을 저장하는 방식으로 결과 테이블 갱신하여 출력
+"""
+from collections import deque
+import copy
+
+
+v = int(input())
+indegree = [0] * (v + 1)
+graph = [[] for i in range(v + 1)]
+time = [0] * (v + 1)
+
+for i in range(1, v + 1):
+    data = list(map(int, input().split()))
+    time[i] = data[0]
+    for x in data[1:-1]:
+        indegree[i] += 1
+        graph[x].append(i)
+
+
+def topology_sort():
+    result = copy.deepcopy(time)
+    q = deque()
+
+    for i in range(1, v + 1):
+        if indegree[i] == 0:
+            q.append(i)
+
+    while q:
+        now = q.popleft()
+        for i in graph[now]:
+            result[i] = max(result[i], result[now] + time[i])
+            indegree[i] -= 1
+
+            if indegree[i] == 0:
+                q.append(i)
+
+    for i in range(1, v + 1):
+        print(result[i])
+
+
+topology_sort()