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
- 트리
- 브루트포스
- 애드 혹
- 수학
- 13164
- 구현
- 2357
- 맵
- JavaScript
- 슬라이딩 윈도우
- 세그먼트 트리
- 누적 합
- 플로이드-워셜
- 싸피
- 그래프
- 모던 JavaScript 튜토리얼
- 그리디
- Python
- 에라토스테네스의 체
- 이분 탐색
- 투 포인터
- SSAFY
- 정수론
- 정렬
- DFS
- boj
- BFS
- 해시 테이블
- DP
- 문자열
Archives
- Today
- Total
흙금이네 블로그
[BOJ] 2887 - 행성 터널 (Python) 본문
아이디어
크루스칼 알고리즘으로 최소 스패닝 트리를 구현하되, 고려하지 않아도 될 터널들은 제외하여 시간을 단축한다.
풀이
좌표 x, y, z 각각을 기준으로 행성들을 정렬하여 인접한 두 행성과 두 행성을 연결하는 데 필요한 비용만을 고려한다.
import sys
input = sys.stdin.readline
def solution():
def find(n):
temp = n
while n != group[n]:
n = group[n]
group[temp] = n
return n
def union(n1, n2):
n1 = find(n1)
n2 = find(n2)
if n1 > n2:
n1, n2 = n2, n1
group[n2] = n1
N = int(input())
planets = []
for i in range(N):
x, y, z = map(int, input().split())
planets.append((x, y, z, i))
group = [i for i in range(N)]
edges = []
for j in range(3):
planets.sort(key=lambda x: x[j])
for i in range(N-1):
d = abs(planets[i+1][j]-planets[i][j])
edges.append((d, planets[i][3], planets[i+1][3]))
edges.sort()
res = cnt = 0
for d, i, j in edges:
if find(i) != find(j):
union(i, j)
res += d
cnt += 1
if cnt >= N-1:
break
print(res)
solution()
'알고리즘' 카테고리의 다른 글
[BOJ] 24526 - 전화 돌리기 (Python) (0) | 2023.04.15 |
---|---|
[BOJ] 2176 - 합리적인 이동경로 (Python) (0) | 2023.04.15 |
[BOJ] 1368 - 물대기 (Python) (0) | 2023.04.15 |
[BOJ] 10423 - 전기가 부족해 (Python) (0) | 2023.04.15 |
[BOJ] 1865 - 웜홀 (Python) (0) | 2023.04.14 |
Comments