행성 연결

• 기본적인 최소 신장 트리 (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