섬 연결하기 - 프로그래머스

섬 연결하기 - 프로그래머스

def solution(n, costs):
    island = [i for i in range(n)]
    costs.sort(key=lambda x: x[2])
    s = set()
    cnt = 0
    s.add(0)
    while len(s) != n:
        for c in costs:
            if c[0] in s or c[1] in s:
                if c[0] in s and c[1] in s:
                    continue
                s.add(c[0])
                s.add(c[1])
                cnt += c[2]
                break
    return cnt