[백준 5972] 택배 배송 C++

문제

백준 5972 택배 배송 C++

5972번: 택배 배송

풀이

문제 설명이 애매하게 되어있는데, 하나 이상의 길로 연결되어 있을 수 있다는 점이 거슬렸다.

시작, 끝 점은 똑같은데 비용이 다른 간선을 말한다기보다는 다른 점으로 우회하여 갈 수 있다고 받아들였다.

 

풀이는 일반적인 다익스트라 문제 해법을 사용해 해결해주면 된다.

우선 순위 큐를 사용하여 시작 점부터 간선 정보를 갱신해 주고, N까지 가는 비용을 출력한다.

소스 코드

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
 
const int MAX = 1987654321;
 
vector<pair<int, int>> graph[50001];
bool visited[50001];
int dist[50001];
int N = 0;
int M = 0;
 
void Dijkstra(int start)
{
    priority_queue<pair<int, int>> pq;
    pq.push(make_pair(0, start));
    dist[start] = 0;
 
    while (!pq.empty())
    {
        int cost = -pq.top().first;
        int cur = pq.top().second;
        pq.pop();
 
        if (visited[cur])
            continue;
 
        visited[cur] = true;
 
        for (size_t i = 0; i < graph[cur].size(); i++)
        {
            int nCost = cost + graph[cur][i].first;
            int next = graph[cur][i].second;
 
            if (nCost < dist[next])
            {
                dist[next] = nCost;
                pq.push(make_pair(-dist[next], next));
            }
        }
    }
}
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
 
    int a = 0;
    int b = 0;
    int c = 0;
    cin >> N >> M;
 
    for (int i = 0; i <= N; i++)
    {
        dist[i] = MAX;
    }
 
    for (int i = 0; i < M; i++)
    {
        cin >> a >> b >> c;
        graph[a].push_back(make_pair(c, b));
        graph[b].push_back(make_pair(c, a));
    }
 
    Dijkstra(1);
    cout << dist[N];
 
    return 0;
}