백준/ Gold 4 문제 , 백준 파이썬 13424 , 비밀 모임 [다익스트라 알고리즘]

백준/ Gold 4 문제 , 백준 파이썬 13424 , 비밀 모임 [다익스트라 알고리즘]

 

 

풀이 시간

 

 

 

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

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

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

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

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

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

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


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


<덧붙일 말>
dist 를 2차원 배열로 두고 풀려고 했는데 어디선가 오류가 발생했는지 틀렸다. 근데 굳이 2차원 할 필요도 없었다 사실... 1차원으로 바꾸고 다시 돌려보니 바로 통과 

 

 

문제 출처

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

13424번: 비밀 모임

 

 

 

 

 

 

정답

 

 

from cmath import inf
import heapq
import sys
input=sys.stdin.readline

for _ in range(int(input())):
    N,M=map(int,input().split())
    graph=[[] for _ in range(N+1)]
    for i in range(M):
        a,b,c=map(int,input().split())
        graph[a].append((c,b))
        graph[b].append((c,a))
        
    f_cnt=int(input())
    friends=list(map(int,input().split()))
    answer=0
    MIN=inf
    
    for j in range(1,N+1):
        q=[(0,j)]
        dist=[inf] * (N+1)
        dist[j]=0
        SUM=0
        while q:
            Z,X=heapq.heappop(q)
            if dist[X]<Z:
                continue
            for cost,next in graph[X]:
                cost=Z+cost
                if cost<dist[next]:
                    dist[next]=cost
                    heapq.heappush(q,[cost,next])
        for k in range(f_cnt):
            SUM+=dist[friends[k]]
            
        if MIN>SUM:
            MIN=SUM
            answer=j
    print(answer)