[알고리즘] 백준 14284번: 간선 이어가기 2 (C++)

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

14284번: 간선 이어가기 2

문제에서 하란대로 간선의 순서까지 조정하며 간선을 추가할 때마다 다익스트라를 돌려야 할 거 같지만 이는 헷갈리게 하려는 의도다.

그냥 다익스트라 한 번 돌리면 끝이다.

#include <iostream>
#include <utility>
#include <queue>
#include <vector>
#define MAX 5000 + 1
#define INF 987654321
#define pii pair<int, int>
using namespace std;

vector<pii> graph[MAX];
int n, m;

struct cmp {
	bool operator()(pii& a, pii& b) {
		return a.second > b.second;
	}
};

vector<int> dijkstra(int node) {
	priority_queue<pii, vector<pii>, cmp> pq;
	vector<int> dist(n + 1, INF);
	dist[node] = 0;
	pq.push({ node, 0 });
	while (!pq.empty()) {
		int now = pq.top().first;
		int cost = pq.top().second;

		pq.pop();
		for (int i = 0; i < graph[now].size(); i++) {
			int next = graph[now][i].first;
			int nextCost = graph[now][i].second;
			if (cost + nextCost < dist[next]) {
				dist[next] = cost + nextCost;
				pq.push({ next, dist[next] });
			}
		}
	}
	return dist;
}

int main() {
	cin >> n >> m;
	for (int i = 0; i < m; i++) {
		int a, b, c;
		cin >> a >> b >> c;
		graph[a].push_back({ b, c });
		graph[b].push_back({ a, c });
	}

	int s, t;
	cin >> s >> t;
	vector<int> temp = dijkstra(s);
	cout << temp[t];
	return 0;
}