13-fnzksxl

[fnzksxl]

🔗 문제 링크

서강그라운드

✔️ 소요된 시간

30분 미만?

✨ 수도 코드

민석님께서 다익스트라 문제 말씀하셔서 먼저 제가 선수쳐봤습니다. 문제 이름부터 깔쌈하길래 혹하고 들어가버렸지 뭐에요 하하ㅏㅅ하

문제를 읽어보시면 금방 눈치채시겠지만, 다익스트라 문제입니다.
어.. 너무 간단해서 딱히 설명할게 없네요.

node가 N개만큼 있을 때, 다익스트라 알고리즘을 N번 돌려봅니다.
각각의 다익스트라 알고리즘을 돌릴 때, 리턴받은 distance 내부 배열이
M보다 작은 숫자일 경우 그 지역의 아이템 개수만큼 더해줍니다.

각각의 결과에 대해서 가장 큰 값을 리턴해주면 정답입니다!

전체 코드

import sys
from heapq import *

input = sys.stdin.readline

def dijkstra(graph, start):
    distance = [float('inf') for _ in graph]
    distance[start] = 0
    queue = []
    heappush(queue, [distance[start], start])

    while queue:
        cur_dist, cur_dest = heappop(queue)

        if distance[cur_dest] < cur_dist:
            continue

        for new_dest, new_dist in graph[cur_dest]:
            dist = cur_dist + new_dist
            if dist < distance[new_dest]:
                distance[new_dest] = dist
                heappush(queue, [dist, new_dest])

    return distance


items = []
N, M, R = map(int, input().split())

item_weight = list(map(int, input().split()))
graph = [[] for _ in range(N+1)]

for _ in range(R):
    start, end, weight = map(int, input().split())
    graph[start].append((end, weight))
    graph[end].append((start, weight))

for item in range(1, len(graph)):
    result = dijkstra(graph, item)
    can_get = 0
    for i in range(len(result)):
        if result[i] <= M:
            can_get += item_weight[i-1]
    items.append(can_get)

print(max(items))

📚 새롭게 알게된 내용

[wkdghdwns199]

태원님 코드 한 번 보고 따라해보고 다익스트라 알고리즘 검색해서 봤는데 기억이 안나고 오히려 어렵네요.. 쉬운 설명 찾으러 떠나봅니다

import sys
from heapq import *
input = sys.stdin.readline

def dijkstra(graph, start):
    length = [float('inf') for _ in graph]
    length[start] = 0
    queue = []
    heappush(queue, [length[start], start])
    
    while queue:
        current_length, current_target = heappop(queue)

        if length[current_target] < current_length :
            continue
        
        for new_target, new_length in graph[current_target] :
            len = current_length + new_length
            if len < length[new_target]:
                length[new_target] = len
                heappush(queue, [len, new_target])
    return length

items = []
N,M,R = map(int, input().split())
T = list(map(int, input().split()))
graph = [[] for _ in range(N+1)]

for _ in range(R):
    start, end, weight = map(int, input().split())
    graph[start].append((end, weight))
    graph[end].append((start, weight))
    
for item in range(1, len(graph)) :
    result = dijkstra(graph, item)
    get = 0
    for i in range(len(result)):
        if result[i] <= M :
            get += T[i-1]
    items.append(get)

print(max(items))