백준 2211번: 네트워크 복구
네트워크 복구
아이디어
문제를 보면 최소 스패닝 트리 문제인가? 싶지만 1번 노드에서 출발할 때 최소 거리를 구하는 문제다. 필요한 간선만 빼고 다 잘라내면, 결국 트리가 된다. 각 노드별로, 자신의 부모 노드가 무엇인지만 기록한다면 어떤 간선을 살려야 하는지 알 수 있다.
코드
#include <bits/stdc++.h>
using namespace std;
constexpr int MAX = 1001, INF = 1e9;
int N, M;
vector<pair<int, int>> adj[MAX]; // cost, node
int dist[MAX], prev_node[MAX];
bool visited[MAX];
void dijkstra() {
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
fill(dist, dist+MAX, INF);
dist[1] = 0;
pq.push({0, 1});
while (!pq.empty()) {
int cost = pq.top().first;
int cur = pq.top().second;
pq.pop();
if (visited[cur]) continue;
visited[cur] = true;
for (auto p : adj[cur]) {
int nc = p.first;
int next = p.second;
if (visited[next]) continue;
if (dist[next] > cost+nc) {
dist[next] = cost+nc;
prev_node[next] = cur;
pq.push({cost+nc, next});
}
}
}
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
cin >> N >> M;
for (int i = 0; i < M; i++) {
int a, b, c;
cin >> a >> b >>c;
adj[a].push_back({c, b});
adj[b].push_back({c, a});
}
dijkstra();
cout << N - 1;
for (int i = 2; i <= N; i++) {
cout << '\n' << i << ' ' << prev_node[i];
}
return 0;
}
여담
문제를 잘 읽도록 하자.
Author And Source
이 문제에 관하여(백준 2211번: 네트워크 복구), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@ks1ksi/백준-2211번-네트워크-복구저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)