Merge pull request #150 from AlgorithmStudy-Allumbus/minjeong

Minjeong / 3월 1주차 / 5문제

Author: learntosurf

_WeeklyChallenges/W13-[Union-Find]/Assignment_BOJ_1717_집합의표현.py

@@ -4,4 +4,36 @@
 유형: Union-Find, Graph
 '''
 
-# PR 올릴 때 과제 정답 공개 예정
+import sys
+sys.setrecursionlimit(10 ** 6)  # 재귀 깊이 제한 늘리기
+input = sys.stdin.readline
+
+# Union 연산(두 집합을 합치기 위한 함수)
+def union(a, b):
+    p_a = find(a)
+    p_b = find(b)
+
+    if p_a > p_b:  # 값이 더 작은 쪽을 부모로 설정
+        parent[p_a] = p_b
+    else:
+        parent[p_b] = p_a
+
+# Find 연산(같은 집합에 속하는지 확인하기 위한 함수)
+def find(a):
+    if a != parent[a]:
+        parent[a] = find(parent[a])  # 경로 압축
+    return parent[a]
+
+# 연산 수행
+n, m = map(int, input().split())
+parent = [i for i in range(n + 1)]  # 초기: 각 원소가 자기 자신을 부모로 가짐
+
+for _ in range(m):
+    flag, a, b = map(int, input().split())
+    if flag == 0:  # 합집합 연산
+        union(a, b)
+    else:  # 같은 집합 확인 연산
+        if find(a) == find(b):
+            print("YES")
+        else:
+            print("NO")

minjeong/Heap/2025-03-08-[백준]-#11279-최대힙.py

@@ -0,0 +1,17 @@
+import sys
+import heapq
+input = sys.stdin.readline
+
+n = int(input())
+heap = []
+
+while n > 0:
+    num = int(input())
+    n -= 1
+    if num == 0:
+        if len(heap) == 0:
+            print(0)
+            continue
+        print(-heapq.heappop(heap))
+    else:
+        heapq.heappush(heap, -num)

minjeong/Heap/2025-03-08-[백준]-#11286-절댓값힙.py

@@ -0,0 +1,16 @@
+import sys
+import heapq
+input = sys.stdin.readline
+
+n = int(input())
+abs_heap = []
+
+for _ in range(n):
+    num = int(input())
+    if num == 0:
+        if abs_heap:
+            print(heapq.heappop(abs_heap)[1])
+        else:
+            print(0)
+    else:
+        heapq.heappush(abs_heap, (abs(num), num))

minjeong/Heap/2025-03-08-[백준]-#1927-최소힙.py

@@ -0,0 +1,17 @@
+import sys
+import heapq
+input = sys.stdin.readline
+
+n = int(input())
+heap = []
+
+for _ in range(n):
+    num = int(input())
+    n -= 1
+    if num == 0:
+        if len(heap) == 0:
+            print(0)
+            continue
+        print(heapq.heappop(heap))
+    else:
+        heapq.heappush(heap, num)

minjeong/UnionFind/2025-03-03-[백준]-#1717-집합의표현.py

@@ -0,0 +1,33 @@
+import sys
+sys.setrecursionlimit(10 ** 6)  # 재귀 깊이 제한 늘리기
+input = sys.stdin.readline
+
+# Union 연산(두 집합을 합치기 위한 함수)
+def union(a, b):
+    p_a = find(a)
+    p_b = find(b)
+
+    if p_a > p_b:  # 값이 더 작은 쪽을 부모로 설정
+        parent[p_a] = p_b
+    else:
+        parent[p_b] = p_a
+
+# Find 연산(같은 집합에 속하는지 확인하기 위한 함수)
+def find(a):
+    if a != parent[a]:
+        parent[a] = find(parent[a])  # 경로 압축
+    return parent[a]
+
+# 연산 수행
+n, m = map(int, input().split())
+parent = [i for i in range(n + 1)]  # 초기: 각 원소가 자기 자신을 부모로 가짐
+
+for _ in range(m):
+    flag, a, b = map(int, input().split())
+    if flag == 0:  # 합집합 연산
+        union(a, b)
+    else:  # 같은 집합 확인 연산
+        if find(a) == find(b):
+            print("YES")
+        else:
+            print("NO")

minjeong/UnionFind/2025-03-03-[백준]-#1976-여행가자.py

@@ -0,0 +1,43 @@
+import sys
+sys.setrecursionlimit(10 ** 8)
+input = sys.stdin.readline
+
+def union(a, b):
+   p_a = find(a)
+   p_b = find(b)
+
+   if p_a > p_b:  # a의 대표보다 b의 대표가 더 작은 값을 가지면,
+       parent[p_a] = p_b  # a가 속한 집합을 b의 집합에 합치기
+   else:
+       parent[p_b] = p_a  # 그렇지 않으면 b의 집합을 a의 집합에 합치기
+
+def find(a):
+   if a == parent[a]:  # a가 자기 자신의 부모이면 대표 노드
+       return a
+
+   parent[a] = find(parent[a])  # 경로 압축을 통해 a의 부모를 대표 노드로 재설정
+   return parent[a]
+
+# 도시의 개수 n과 여행 계획에 포함된 도시의 수 m을 입력받기
+n = int(input())
+m = int(input())
+
+parent = [i for i in range(n)]  # 각 도시는 처음에 자기 자신이 대표 노드.
+
+# n개의 줄에 걸쳐 도시 간 연결 정보를 입력받고, 연결되어 있으면 union 연산 수행
+for i in range(n):
+   arr = list(map(int, input().split()))
+   for j in range(n):
+       if arr[j]:  # 1이면 i와 j가 연결되어 있으므로 union 연산 수행
+           union(i, j)
+
+# 여행 계획을 입력
+plan = list(map(int, input().split()))
+result = "YES"
+# 여행 계획에 있는 모든 도시가 같은 집합(대표 노드)을 가지는지 확인
+for i in range(1, m):
+   if parent[plan[i]-1] != parent[plan[0]-1]:
+       result = "NO"
+       break
+
+print(result)