백준/ Gold 3 문제 , 백준 파이썬 16202 , MST 게임 [최소 스패닝 트리, 프림 알고리즘]

백준/ Gold 3 문제 , 백준 파이썬 16202 , MST 게임 [최소 스패닝 트리, 프림 알고리즘]

 

 

풀이 시간

 

 

 

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

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

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

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

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

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

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


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


<덧붙일 말>
원래 리스트 를 보존하기 위해 deepcopy 사용

 

 

 

문제 출처

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

16202번: MST 게임

 

 

 

 

 

 

 

정답

 

 

 

import heapq
import copy
import sys
input=sys.stdin.readline
    
def MST(a,graph):
    visited=[True,True]+[False]*(N-1)
    tmp=copy.deepcopy(graph)
    q=tmp[1]
    heapq.heapify(q)
    result=0
    cnt=0
    while q:
        Z,X=heapq.heappop(q)
        if Z<=a:
            continue
        if not visited[X]:
            visited[X]=True
            result+=Z
            cnt+=1
            for i in tmp[X]:
                if not visited[i[1]]:
                    heapq.heappush(q,i)
    if cnt==N-1:
        return result
    else:
        return 0

N,M,T=map(int,input().split())
graph=[[] for _ in range(N+1)]
for i in range(M):
    a,b=map(int,input().split())
    graph[a].append((i+1,b))
    graph[b].append((i+1,a))

answer=[]
for i in range(T):
    result=MST(i,graph)
    if result==0:
        break
    answer.append(result)

while len(answer)<T:
    answer.append(0)

print(*answer)