본문 바로가기

백준

22116 창영이와 퇴근

풀이 자체는 쉬웠다.

인접 배열을 전부 간선으로 만들고 또 MST 돌려야지. 

그리고! 전에 북님이 알려준대로 find(s) == find(e)인 순간이 최대 경사의 최솟값이 돼서 종료하면 되겠다 싶었다. 

 

그런데 한 열 번은 틀렸다...........

인덱스를 따로 관리하니 너무너무너무 헷갈렸다......

틀린 이유

1. 첫 번째 시도

노드 번호를 <i,j> 형식으로 붙이기 어려워서 이걸 전부 인덱스를 따로 붙였다.

0 1 2 

3 4 5

6 7 8

이런 식으로

 

그러면 인덱스가 n * n - 1 사이즈까지 가는데, parent 배열의 사이즈는 n까지 잡았다. 

그래서 out of bound 에러가 떴다.

 

그 이후 또 찾아보려고 했는데 도저히 원인을 몰라서 울고 싶었다. 

알고보니...

 

2. 두 번째 시도

로직 문제와 int 범위 벗어난 문제는 없는 것 같고 내 눈으로 도저히 못 찾겠어서 정석적으로 원인을 찾아보려고 했다. 

나의 정석 해결 방안

- 최솟값을 넣는다

- 최댓값을 넣는다

- 하........... 랜덤 값 좀 넣어보고 더 고민해본다

 

그런데 최솟값인 n=1을 넣으니 출력이 아무것도 안 됐다!

경사가 없는 경우라 0을 출력하고 종료했다. 

 

통과~~

#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>;

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

int n;
int par[1000005];
int mapv[1005][1005];

void Init() {
    for (int i = 0; i <= n * 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 convertIndex(int x, int y) {
    return n * x + y;
}

ii convertValue(int idx) {
    ii res;
    res.xx = idx / n;
    res.yy = idx % n;
    return res;
}

int main() 
{
    scanf("%d", &n);
    if (n == 1) {
        printf("0\n");
        return 0;
    }
    
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            scanf("%d", &mapv[i][j]);
    
    Init();
    vector<Edge> edge;
    for (int i = 0; i < n; i++) {
        for (int j = 1; j < n; j++) {
            Edge e;
            e.u = convertIndex(i, j - 1);
            e.v = convertIndex(i, j);
            e.d = abs(mapv[i][j] - mapv[i][j - 1]);
            edge.push_back(e);
            
            e.u = convertIndex(j - 1, i);
            e.v = convertIndex(j, i);
            e.d = abs(mapv[j][i] - mapv[j - 1][i]);
            edge.push_back(e);
        }
    }
    
    sort(all(edge), [](const Edge &l, const Edge &r) {
        return l.d < r.d;
    });
    

    for (auto e: edge) {
        if (Find(e.u) == Find(e.v))
            continue;
        Merge(e.u, e.v);
        if (Find(convertIndex(0, 0)) == Find(convertIndex(n-1, n-1))) {
            printf("%d\n", e.d);
            return 0;
        }
    }
    
    return 0;
}

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

20026 Fix Wiring  (0) 2022.08.23
2406 안정적인 네트워크  (0) 2022.08.21
16398 행성 연결  (0) 2022.08.20
13905 세부  (0) 2022.08.10
1647 도시 분할 계획  (0) 2022.08.09