백준/ Gold 4 문제 , 백준 파이썬 14938 , 서강그라운드 [다익스트라 알고리즘]

백준/ Gold 4 문제 , 백준 파이썬 14938 , 서강그라운드 [다익스트라 알고리즘]

 

 

풀이 시간

 

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

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

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

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

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

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

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


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


<덧붙일 말>
방문 노드들을 확인하고 방문한 곳들만 값 계산을 해주는 식으로 했음.
dist 를 사용하지 않고 heap 에 넣어지는 값들만 더해서 MAX 값을 계산하면 예제는 잘 나오지만 12%에서 계속 틀림.

 

문제 출처

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

14938번: 서강그라운드

 

 

 

 

 

 

정답

 

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

n,m,r=map(int,input().split())
arr=[0]+list(map(int,input().split()))
graph=[[] for _ in range(n+1)]

for i in range(r):
    a,b,c=map(int,input().split())
    graph[a].append((c,b))
    graph[b].append((c,a))

MAX=0
for i in range(1,n+1):
    dist=[inf]*(n+1)
    q=[(i,0)]
    dist[i]=0
    result=arr[i]
    while q:
        node,cost=heapq.heappop(q)
        if dist[node]<cost:
            continue
        for C,next in graph[node]:
            tmp=C+cost
            if tmp<=m:
                if tmp<=dist[next]:
                    dist[next]=tmp
                    result+=arr[next]
                    heapq.heappush(q,[next,tmp])
    SUM=0
    for j in range(1,n+1):
        if dist[j]<=m:
            SUM+=arr[j]
    if MAX<SUM:
        MAX=SUM
                        
print(MAX)