백준/ Gold 4 문제 , 백준 파이썬 1647 , 도시 분할 계획 [최소 스패닝 트리]

백준/ Gold 4 문제 , 백준 파이썬 1647 , 도시 분할 계획 [최소 스패닝 트리]

 

 

풀이 시간

 

 

Check Point !   ( 해당사항 ✓체크 )

1. 막힘 없이 수월하게 풀린 문제인가?

2. 1시간이내로 풀렸던 문제인가?

3. 1시간 이상 or 며칠을 두고 풀어봤더니 풀린 문제인가?

4. 시간을 써도 도무지 풀 수 없는 문제인가?

5. 솔루션을 찾아봤는가?

-------------------------------------------------------------------------------------------
난이도 체감
1. 최상
2. 상
3. 중
4. 하


<이해도>
1. 완벽히 이해
2. 다소 헷갈리는 부분들이 있음
3. 이해 못함


<덧붙일 말>
문제속에서 요구하는 해답이 정확히 무엇인지 첨에 잘 이해를 못했다.
그러다가 '나머지 길의 유지비의 합을 최소로 하고 싶다' 라는 말이 결국 최소 가중치로 다 이은 다음 그중 가장 큰 가중치의 값을 제거 해야 한다는 뜻인걸 이해. 가장 큰 가중치 하나를 빼주니 맞았다.

 

 

문제 출처

https://www.acmicpc.net/problem/1647

1647번: 도시 분할 계획

 

 

 

 

 

풀이

 

프림 알고리즘을 적용하였고 최소 가중치들을 모두 이어준 뒤 결과 값에서 가장 큰 값을 빼준다.

 

이때 tmp 변수를 사용해서 가장 큰 값을 할당하고 마지막 출력때 빼줘서 출력시키면 된다.

 

 

정답

 

import sys
input=sys.stdin.readline
import heapq

N,M=map(int,input().split())
graph=[[] for _ in range(N+1)]
visited=[True,True]+[False]*(N-1)
for i in range(M):
    a,b,c=map(int,input().split())
    graph[a].append((c,b))
    graph[b].append((c,a))
    
q=graph[1]
heapq.heapify(q)
result=0

tmp=0
while q:
    Z,X=heapq.heappop(q)
    if not visited[X]:
        visited[X]=True
        tmp=max(tmp,Z)
        result+=Z
        for i in graph[X]:
            if not visited[i[1]]:
                heapq.heappush(q,i)

print(result-tmp)