프로그래머스 섬 연결하기 파이썬(PYTHON)

문제링크 : programmers.co.kr/learn/courses/30/lessons/42861

코딩테스트 연습 - 섬 연결하기

 

 

def solution(n, costs):
    answer = 0
    startnum = [0]
    visitland=[]
    cost = list(costs)
    cost.sort(key= lambda x : x[2])
    visitland.append(cost[0][0])
    visitland.append(cost[0][1])
    answer+=cost[0][2]
    cost.pop(0)
    while len(visitland) != n:
        for i in cost:
            if i[0] in visitland or i[1] in visitland:
                if i[0] in visitland and i[1] in visitland:
                    continue
                else :
                    if i[0] not in visitland:
                        visitland.append(i[0])
                    if i[1] not in visitland:
                        visitland.append(i[1])
                    answer+=i[2]
                    break
    return answer