티스토리 뷰

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

 

1504번: 특정한 최단 경로

첫째 줄에 정점의 개수 N과 간선의 개수 E가 주어진다. (2 ≤ N ≤ 800, 0 ≤ E ≤ 200,000) 둘째 줄부터 E개의 줄에 걸쳐서 세 개의 정수 a, b, c가 주어지는데, a번 정점에서 b번 정점까지 양방향 길이 존�

www.acmicpc.net

백준 소스코드 [C++] 1504 특정한 최단 경로

#include <iostream>
#include <algorithm>
#include <queue>
#include <string.h>
#include <limits.h>
using namespace std;
#define MAX 987654321
int v, e, s, d, weight;
vector <pair<int, int>> adj[800];
vector<int> dijkstra(int src) {
	priority_queue <pair<int, int> > pq;
	vector<int> dist(v, 987654321);
	dist[src] = 0;
	pq.push({ 0,src });
	while (pq.empty() != true) {
		int cost = -pq.top().first;
		int here = pq.top().second;
		pq.pop();
		if (dist[here] < cost) continue;
		for (int i = 0;i < (int)adj[here].size();i++) {
			int there = adj[here][i].first;
			int nextdist = cost + adj[here][i].second;
			if (dist[there] > nextdist) {
				dist[there] = nextdist;
				pq.push({ -nextdist,there });
			}
		}
	}
	return dist;
}
int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cin >> v >> e;
	for (int i = 0;i < e;i++) {
		cin >> s >> d >> weight;
		adj[s - 1].push_back(make_pair(d - 1, weight));
		adj[d - 1].push_back(make_pair(s - 1, weight));
	}
	cin >> s >> d;
	long long res = min(dijkstra(1 - 1)[s - 1] + dijkstra(s - 1)[d - 1] + dijkstra(d - 1)[v - 1],
		dijkstra(1 - 1)[d - 1] + dijkstra(d - 1)[s - 1] + dijkstra(s - 1)[v - 1]);
	if (res > 987654321 || e==0) cout << "-1";
	else cout << res;
	return 0;
}
댓글