[파이썬] 프로그래머스 섬 연결하기

잘못 푼 줄 알고 해설을 봤는데 내가 풀던 거랑 똑같아서 다시 풀어보니 맞았다.

def solution(n, costs):
    answer = 0
 
    costs.sort(key=lambda x:x[2])
    visited = [0]*n
 
    visited[costs[0][0]]=1
 
    while(1):
        if sum(visited)==n:
            break
        else:
            for cost in costs:
                a,b,c = cost
 
                if visited[a]==1 and visited[b]==1:
                    continue
                if visited[a]+visited[b]==1:
                    visited[a]=1
                    visited[b]=1
                    answer+=c
                    break
 
    return answer