본문 바로가기

백준

16398 행성 연결

이 문제 보자마자 그냥 행렬로 들어온 MST구나~ 싶었다.

그런데 순간 Cij의 최댓값이 10^9이라 어 숫자가 큰데?? 그런데 N은 1000이라 좌표압축이 필요하겠다 싶었다. 

(왜 이렇게 생각했지? 퇴근하고 바로 풀어서 그랬나보다)

 

그렇게 생각만 하고 일주일 뒤 오늘 바로 좌표압축부터 공부했다.

직접 손코딩하면서 또 코드 안 보고 짤 수 있도록 익혔다. 그러곤 문제 다시 읽으니 좌표압축이 굳이 필요 없는 문제였다... 

 

해결~

 

#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 mapv[1005][1005];

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

void Init(int n) {
    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() {
    int n;
    scanf("%d", &n);
    
   vector<Edge> edge;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            scanf("%d", &mapv[i][j]);
            if (j < i) {
                Edge e;
                e.u = i;
                e.v = j;
                e.d = mapv[i][j];
                edge.push_back(e);
            }
        }
    }
    
    Init(n);
    
    sort(all(edge), [](const Edge &l, const Edge &r) {
       return l.d < r.d; 
    });
    
    i64 ans = 0;
    for (auto e: edge) {
        if (Find(e.u) == Find(e.v))
            continue;
        
        ans += e.d;
        Merge(e.u, e.v);
    }
    
    printf("%lld", ans);
    
    
    return 0;
}

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

2406 안정적인 네트워크  (0) 2022.08.21
22116 창영이와 퇴근  (0) 2022.08.21
13905 세부  (0) 2022.08.10
1647 도시 분할 계획  (0) 2022.08.09
1197 최소 스패닝 트리  (0) 2022.08.08