Tsinsen A1039bzoj2638黑白染色 （BFS树）

Posted 2021-01-10 lstete

Descroption

原题链接 你有一个(n*m)的矩形，一开始所有格子都是白色，然后给出一个目标状态的矩形，有的地方是白色，有的地方是黑色，你每次可以选择一个连通块（四连通块，且不要求颜色一样）进行染色操作（染成白色或者黑色）。问最少操作次数。(1 leq n, m, leq 50.)

Solution

对目标矩形的同色联通块缩点，向相邻的异色联通块连边，代表先把该块和它所有相邻的块染成同色，再把该块染成异色。至于为什么是最优的我也不知道，考场上手玩出来的qwq那么这个图的某一颗(BFS)树的最大深度就是以该树根为染色中心向四周扩展的答案，又因为范围很小，考虑枚举树根。

Code

#include <bits/stdc++.h>
#define Set(a, b) memset(a, b, sizeof (a))
#define For(i, j, k) for (register int i = j; i <= k; ++i)
#define Forr(i, j, k) for (register int i = j; i >= k; --i)
using namespace std;

inline void File() {
    freopen("bzoj2638.in", "r", stdin);
    freopen("bzoj2638.out", "w", stdout);
}

const int N = 50 + 5, M = N * N; char s[N];
const int dir[4][2] = {-1, 0, 0, -1, 1, 0, 0, 1};
struct node { int sx, sy; } st[M];
int a[N][N], n, m, vis[N][N], dep[M];
int id[M], cnt, ans, tt[M], vis1[N][N];

vector<int> nxt[M];

inline void dfs(int x, int y) {
    vis[x][y] = cnt, id[cnt] = a[x][y];
    st[cnt] = (node) {x, y};
    For(i, 0, 3) {
        int dx = x + dir[i][0], dy = y + dir[i][1];
        if (dx < 1 || dx > n || dy < 1 || dy > m || a[x][y] ^ a[dx][dy] || vis[dx][dy]) continue;
        dfs(dx, dy);
    }
} 

inline void dfs1(int x, int y) {
    vis1[x][y] = 1;
    For(i, 0, 3) {
        int dx = x + dir[i][0], dy = y + dir[i][1];
        if (dx < 1 || dx > n || dy < 1 || dy > m || vis1[dx][dy]) continue;
        if (vis[x][y] ^ vis[dx][dy]) {
            if (!tt[vis[dx][dy]]) tt[vis[dx][dy]] = 1, nxt[vis[x][y]].push_back(vis[dx][dy]);
            continue;
        }
        dfs1(dx, dy);
    }
} 

inline void Solve(int x) {

    Set(tt, 0); queue<int> Q;
    Q.push(x), tt[x] = 1, dep[x] = 1;
    
    while (!Q.empty()) {
        int u = Q.front(); Q.pop();
        for (int i = 0, sz = nxt[u].size(); i < sz; ++ i) {
            int v = nxt[u][i];
            if (!tt[v]) Q.push(v), dep[v] = dep[u] + 1, tt[v] = 1; 
        }   
    }

    int res = 0, p = 0;
    For(i, 1, cnt) if (dep[i] > res) res = dep[i], p = i;
    ans = min(ans, res - (id[p] == 0));
}

int main() {
    File();

    cin >> n >> m;
    For(i, 1, n) {
        scanf("%s", s + 1);
        For(j, 1, m) a[i][j] = s[j] == ‘B‘;
    }

    For(i, 1, n) For(j, 1, m) if (!vis[i][j]) ++ cnt, dfs(i, j);
    For(t, 1, cnt) Set(vis1, 0), Set(tt, 0), dfs1(st[t].sx, st[t].sy);
    
    ans = n * m + 10;
    For(i, 1, cnt) Solve(i);
    printf("%d
", ans);
    return 0;
}