https://www.acmicpc.net/problem/9370
이 문제는 처음에 플로이드로 풀다가 시간초과 나서 djikstra로 변경하였다 이문제는 그냥 다익스트라를 세번 돌리면 되는데
int route1 = dist_s[g] + g_h_cost + dist_h[dest]; // s -> g -> h -> dest
int route2 = dist_s[h] + g_h_cost + dist_g[dest]; // s -> h -> g -> dest
이렇게 실제 거리와 g_h를 지나는 거리와 똑같다면 g -> h를 지나간다고 판별 할 수 있으니
이를 이용해서 목적지까지의 거리가 같으면 priority_queue에 넣은 후 차례로 pop하면 되는 문제였다
#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
#define INF 1000000001
int t, n, m, d;
int s, g, h;
vector<pair<int, int>> edgeList[2001];
vector<int> distance_v;
void dijkstra(int start, vector<int>& dist) {
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
dist.assign(n + 1, INF);
dist[start] = 0;
pq.push({ 0, start });
while (!pq.empty()) {
int cur_cost = pq.top().first;
int cur_vertex = pq.top().second;
pq.pop();
if (cur_cost > dist[cur_vertex])
continue;
for (auto next : edgeList[cur_vertex]) {
int next_vertex = next.first;
int next_cost = next.second;
if (dist[next_vertex] > cur_cost + next_cost) {
dist[next_vertex] = cur_cost + next_cost;
pq.push({ dist[next_vertex], next_vertex });
}
}
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin >> t;
while (t--) {
cin >> n >> m >> d;
cin >> s >> g >> h;
for (int i = 0; i <= n; i++) {
edgeList[i].clear();
}
int g_h_cost = 0;
for (int i = 0; i < m; i++) {
int from, to, cost;
cin >> from >> to >> cost;
edgeList[from].push_back({ to, cost });
edgeList[to].push_back({ from, cost });
if ((from == g && to == h) || (from == h && to == g)) {
g_h_cost = cost; // g-h 간선의 비용 저장
}
}
// 다익스트라 실행
vector<int> dist_s(n + 1), dist_g(n + 1), dist_h(n + 1);
dijkstra(s, dist_s);
dijkstra(g, dist_g);
dijkstra(h, dist_h);
priority_queue<int, vector<int>, greater<int>> destination_pq;
for (int i = 0; i < d; i++) {
int dest;
cin >> dest;
int route1 = dist_s[g] + g_h_cost + dist_h[dest]; // s -> g -> h -> dest
int route2 = dist_s[h] + g_h_cost + dist_g[dest]; // s -> h -> g -> dest
if (dist_s[dest] == route1 || dist_s[dest] == route2) {
destination_pq.push(dest);
}
}
// 결과 출력
while (!destination_pq.empty()) {
cout << destination_pq.top() << " ";
destination_pq.pop();
}
cout << endl;
}
}
'백준(코테준비) > DP' 카테고리의 다른 글
백준 1719 / C++ / 다익스트라 / DP (0) | 2025.02.13 |
---|---|
백준 1937 / C++ / dp (0) | 2025.02.07 |
백준 1562 / C++ / DP / 비트마스킹 (0) | 2025.02.07 |
백준 9252 / C++ / Dp (0) | 2025.01.24 |
백준 17404 / CPP / Dp (0) | 2025.01.16 |