섬 연결하기

ps
링크programmers.co.kr/…
출처프로그래머스
문제 번호42861
문제명섬 연결하기
레벨Level 3
분류

최소 신장 트리

시간복잡도O(ElogV)
인풋사이즈V<=100, E<=5000
사용한 언어Python
해결날짜2020/11/23
태그

고득점 Kit - 탐욕법

코드

"""Solution code for "Programmers 42861. 섬 연결하기".

- Problem link: https://programmers.co.kr/learn/courses/30/lessons/42861
- Solution link: http://www.teferi.net/ps/problems/programmers/42861
"""

from teflib import tgraph


def solution(n, costs):
    wgraph = [{} for _ in range(n)]
    for u, v, w in costs:
        wgraph[u][v] = wgraph[v][u] = w
    return tgraph.minimum_spanning_tree(wgraph)