네트워크 연결 (feat. 백준, 1922번)

네트워크 연결
https://www.acmicpc.net/problem/1922

1922번: 네트워크 연결

 

// woohyeon.kim
// kim519620.tistory.com
#include <iostream>
#include <string.h>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
bool visit[1001];
vector<pair<int, int>> arr[1001];
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> q;
int main(){
    ios::sync_with_stdio(false), cin.tie(NULL);
    register int N, M, ans = 0;
    cin >> N >> M;
    memset(visit, false, N);
    for(register int m = 0, a, b, c; m < M; ++m){
        cin >> a >> b >> c;
        arr[a].push_back(make_pair(b, c)), arr[b].push_back(make_pair(a, c));
    }
    q.push(make_pair(0, 1));
    register int next, nnext, cost;
    while(!q.empty()){
        next = q.top().second, cost = q.top().first; q.pop();
        if(visit[next])
            continue;
        visit[next] = true;
        ans += cost;
        for(auto& v : arr[next]){
            nnext = v.first;
            if(visit[nnext])
                continue;
            cost = v.second;
            q.push(make_pair(cost, nnext));
        }
    }
    cout << ans;
	return 0;
}
// *&)*@*

 

반응형

최소 스패닝 트리 (MST) 를 구하는 문제로 크루스칼 알고리즘 또는 프림 알고리즘을 통해 손쉽게 구할 수 있습니다.