[BOJ] 백준 16234 - 인구 이동

문제 링크

www.acmicpc.net/problem/16234

16234번: 인구 이동

 

풀이

1) 아직 방문한 적 없는 칸에서 dfs를 돌려 갈 수 있는 곳까지 간 후, 인구수의 평균을 구해줍시다.

2) 그 칸에서 다시 한번 dfs를 돌려 각 나라의 인구수를 1에서 구한 평균값으로 바꿔줍니다.

3) 더이상 인구수의 변화가 없을 때까지 반복해주면 됩니다.

 

정답 코드

#include <bits/stdc++.h>

#define TEST int _tt; cin >> _tt; while(_tt--)
#define ft first
#define sd second
#define all(x) (x).begin(), (x).end()
#define rmDup(x) (x).erase(unique((x).begin(), (x).end()), (x).end())
#define mem(v, e) memset((v), (e), sizeof((v)))
#define initme(p, n) for(int _i=0; _i<(n)+1; (p)[_i]=_i, _i++)
using namespace std;
using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using vi = vector<int>;
using vb = vector<bool>;
using vs = vector<string>;
using vd = vector<double>;
using vll = vector<ll>;
using vpii = vector<pii>;
using vpll = vector<pll>;
using vvi = vector<vi>;
using vvb = vector<vb>;
using vvll = vector<vll>;
using vvpii = vector<vpii>;
using vvpll = vector<vpll>;
using vvs = vector<vs>;
const int INF = 0x3f3f3f3f;
const ll LLINF = 0x3f3f3f3f3f3f3f3fLL;

int n, l, r;
bool c = true;
int p[55][55];
int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0};
bool chk[55][55], chk2[55][55];

bool isValidNode(int x, int y) {
    return 0 <= x && x < n && 0 <= y && y < n;
}

bool isValid(int x, int y, int x2, int y2) {
    int val = abs(p[x][y] - p[x2][y2]);
    return l <= val && val <= r;
}

pii dfs(int x, int y) {
    pii ret = {1, p[x][y]};
    chk[x][y] = true;
    for (int k = 0; k < 4; k++) {
        int nx = x + dx[k], ny = y + dy[k];
        if (!isValidNode(nx, ny) || !isValid(x, y, nx, ny) || chk[nx][ny]) continue;
        pii val = dfs(nx, ny);
        ret.ft += val.ft;
        ret.sd += val.sd;
    }
    return ret;
}

void dfs2(int x, int y, int val) {
    chk2[x][y] = true;
    for (int k = 0; k < 4; k++) {
        int nx = x + dx[k], ny = y + dy[k];
        if (!isValidNode(nx, ny) || !isValid(x, y, nx, ny) || chk2[nx][ny]) continue;
        dfs2(nx, ny, val);
    }
    p[x][y] = val;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    cin >> n >> l >> r;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            cin >> p[i][j];
        }
    }
    int ans = -1;
    while (c) {
        c = false;
        ans++;
        mem(chk, 0);
        mem(chk2, 0);
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (!chk[i][j]) {
                    auto[cnt, val] = dfs(i, j);
                    if (cnt > 1) {
                        int newVal = val / cnt;
                        dfs2(i, j, newVal);
                        c = true;
                    }
                }
            }
        }
    }
    cout << ans;
}