ps:problems:boj:16398
행성 연결
ps | |
---|---|
링크 | acmicpc.net/… |
출처 | BOJ |
문제 번호 | 16398 |
문제명 | 행성 연결 |
레벨 | 골드 4 |
분류 |
최소 신장 트리 |
시간복잡도 | O(n^2) |
인풋사이즈 | n<=1000 |
사용한 언어 | Python |
제출기록 | 72268KB / 596ms |
최고기록 | 516ms |
해결날짜 | 2021/10/24 |
풀이
- 기본적인 최소 신장 트리 (Minimum Spanning Tree / MST) 문제. 그냥 알고리즘 적용해서 계산하면 된다
- 주어진 그래프가 완전 그래프인만큼, 일반적으로 사용하는 O(ElogV)의 크루스칼 알고리즘보다 O(V^2)의 프림 알고리즘을 쓰는 것이 더 효율적이다
코드
"""Solution code for "BOJ 16398. 행성 연결".
- Problem link: https://www.acmicpc.net/problem/16398
- Solution link: http://www.teferi.net/ps/problems/boj/16398
Tags: [MST]
"""
import sys
from teflib import tgraph
def main():
n = int(sys.stdin.readline())
weight_mat = [
[int(x) for x in sys.stdin.readline().split()] for _ in range(n)
]
print(tgraph.prim_mst(n, lambda u, v: weight_mat[u][v]))
if __name__ == '__main__':
main()
- Dependency: teflib.tgraph.prim_mst
ps/problems/boj/16398.txt · 마지막으로 수정됨: 2021/10/24 04:46 저자 teferi
토론