본문 바로가기

백준

2406 안정적인 네트워크

문제 처음 읽었을 때는 바로 이해 안 됐지만 그려보니 바로 이해됐다.

일단 집합과 집합을 연결하는 최소 비용을 고르면 된다. 

그래서 먼저 같은 집합으로 묶어놓고 같은 집합이라면 패스하도록 구현했다. 

 

 

 

 

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <deque>
#include <stack>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <bitset>
#include <stdio.h>
#include <math.h>
#include <sstream>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()

using namespace std;
using i64 = long long int;
using ii = pair<int, int>;
using iis = pair<int, string>;
using ii64 = pair<i64, i64>;
using iii = tuple<int, int, int>;

int par[1005];
int n, m;

struct Edge {
    int u, v, d;
};

void Init() {
    for (int i = 1; i <= n; i++)
        par[i] = i;
}

int Find(int x) {
    if (x == par[x])
        return x;
    return par[x] = Find(par[x]);
}

void Merge(int x, int y) {
    x = Find(x);
    y = Find(y);
    par[x] = y;
}

int main() {
    scanf("%d %d", &n, &m);
    
    Init();
    for (int i = 0; i < m; i++) {
        int x, y;
        scanf("%d %d", &x, &y);
        Merge(x, y);
    }
    
    vector<Edge> edge;
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            int x;
            scanf("%d", &x);
            
            if (i == 1 || j == 1 || i <= j)
                continue;
            
            Edge e;
            e.u = i;
            e.v = j;
            e.d = x;
            
            if (Find(e.u) == Find(e.v))
                continue;
            
            edge.push_back(e);
        }
    }
    
    sort(all(edge), [](const Edge &l, const Edge &r) {
        return l.d < r.d;
    });
    
    i64 price = 0;
    vector<ii> ans;
    for (auto e: edge) {
        if (Find(e.u) == Find(e.v))
            continue;
        
        price += e.d;
        ans.emplace_back(e.u, e.v);
        
        Merge(e.u, e.v);
    }
    
    printf("%lld %d\n", price, ans.size());
    for (int i = 0; i < ans.size(); i++) {
        printf("%d %d\n", ans[i].xx, ans[i].yy);
    }
    
    
    return 0;
}

 

 

'백준' 카테고리의 다른 글

11995 Fenced In (Gold) [미완]  (0) 2022.08.24
20026 Fix Wiring  (0) 2022.08.23
22116 창영이와 퇴근  (0) 2022.08.21
16398 행성 연결  (0) 2022.08.20
13905 세부  (0) 2022.08.10