Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
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 |
Tags
- 2357
- 세그먼트 트리
- 누적 합
- 투 포인터
- BFS
- 플로이드-워셜
- 에라토스테네스의 체
- 문자열
- 구현
- 이분 탐색
- 트리
- 정수론
- 그래프
- Python
- 애드 혹
- 그리디
- 해시 테이블
- 모던 JavaScript 튜토리얼
- DP
- 13164
- 슬라이딩 윈도우
- 맵
- SSAFY
- 수학
- 정렬
- 브루트포스
- DFS
- boj
- 싸피
- JavaScript
Archives
- Today
- Total
흙금이네 블로그
[BOJ] 16562 - 친구비 (Python) 본문
아이디어
유니온 파인드를 이용하여 상호 배타적 집합의 개수를 구하고, 집합 내 최소 비용의 합을 구한다.
풀이
import sys
input = sys.stdin.readline
def solution():
for _ in range(M):
v, w = map(int, input().split())
union(v, w)
for i in range(1, N+1):
find(i)
res = 0
for i in set(group):
res += A[i]
if res <= k:
print(res)
else:
print('Oh no')
def find(n):
temp = n
while n != group[n]:
n = group[n]
group[temp] = n
return n
def union(v, w):
v = find(v)
w = find(w)
if v > w:
v, w = w, v
group[w] = v
if A[w] < A[v]:
A[v] = A[w]
N, M, k = map(int, input().split())
A = [0]+list(map(int, input().split()))
group = [i for i in range(N+1)]
solution()
'알고리즘' 카테고리의 다른 글
[BOJ] 14499 - 주사위 굴리기 (Python) (0) | 2023.03.16 |
---|---|
[BOJ] 1351 - 무한 수열 (Python) (0) | 2023.03.16 |
[BOJ] 27724 - 팝핀 소다 (Python) (0) | 2023.03.16 |
[BOJ] 7662 - 이중 우선순위 큐 (Python) (0) | 2023.03.12 |
[BOJ] 5547 - 일루미네이션 (Python, JavaScript) (0) | 2023.03.11 |
Comments