BOJ)2211 네트워크 복구

문제:icpc.me/2211

원본 네트워크가 주어질 때 1.최소 개수의 회선만 복구하면서 2. 1번 노드에서 k번 노드의 최단거리가 원본보다 길어지지 않도록 복구하라는 문제이다.

처음에 1번 조건만 봤을때는 MST를 떠올릴 수 있으나 2번조건 때문에 다익스트라를 이용하여 1번 정점부터 최단거리를 구해주면서 사용되는 모든 간선을 벡터에 저장해주면 된다.

#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
#define MAX_N 1000
using namespace std;
int n, m, a, b, c, dp[MAX_N + 1];
priority_queue<pair<int, pair<int, int>>> pq;
vector<pair<int, int>> res;
vector<vector<pair<int, int>>> vt;
int main() {
    scanf("%d%d", &n, &m);
    vt.resize(n + 1);
    for (int i = 0; i < m; i++) {
        scanf("%d%d%d", &a, &b, &c);
        vt[a].push_back({ b,c });
        vt[b].push_back({ a,c });
    }
    memset(dp, -1, sizeof(dp));
    pq.push({ 0,{1,0} });
    while (pq.size()) {
        int here = pq.top().second.first;
        int cost = -pq.top().first;
        int par = pq.top().second.second;
        pq.pop();
        if (dp[here] != -1)continue;
        dp[here] = cost;
        if (par)
            res.push_back({ here,par });
        for (auto next:vt[here]) {
            int there = next.first;
            if (dp[there] != -1)continue;
            pq.push({ -cost - next.second, {there,here} });
        }
    }
    printf("%d\n", res.size());
    for (auto x : res)
        printf("%d %d\n", x.first, x.second);
    return 0;
}