洛谷P1162填涂颜色

填涂颜色

题目描述

由数字 $0$ 组成的方阵中，有一任意形状的由数字 $1$ 构成的闭合圈。现要求把闭合圈内的所有空间都填写成 $2$。例如：$6\times 6$ 的方阵（$n=6$），涂色前和涂色后的方阵如下：

如果从某个 $0$ 出发，只向上下左右 $4$ 个方向移动且仅经过其他 $0$ 的情况下，无法到达方阵的边界，就认为这个 $0$ 在闭合圈内。闭合圈不一定是环形的，可以是任意形状，但保证闭合圈内的 $0$ 是连通的（两两之间可以相互到达）。

0 0 0 0 0 0
0 0 0 1 1 1
0 1 1 0 0 1
1 1 0 0 0 1
1 0 0 1 0 1
1 1 1 1 1 1

0 0 0 0 0 0
0 0 0 1 1 1
0 1 1 2 2 1
1 1 2 2 2 1
1 2 2 1 2 1
1 1 1 1 1 1

输入格式

每组测试数据第一行一个整数 $n(1\le n\le 30)$。

接下来 $n$ 行，由 $0$ 和 $1$ 组成的 $n\times n$ 的方阵。

方阵内只有一个闭合圈，圈内至少有一个 $0$。

输出格式

已经填好数字 $2$ 的完整方阵。

样例 #1

样例输入 #1

6
0 0 0 0 0 0
0 0 1 1 1 1
0 1 1 0 0 1
1 1 0 0 0 1
1 0 0 0 0 1
1 1 1 1 1 1

样例输出 #1

0 0 0 0 0 0
0 0 1 1 1 1
0 1 1 2 2 1
1 1 2 2 2 1
1 2 2 2 2 1
1 1 1 1 1 1

提示

对于 $100\%$ 的数据，$1\le n\le 30$。

思路：
用bfs来解决。
把所有的数通过bfs遍历一遍，把与边界相连的0换成7。
首先从(0，0)开始，bfs一遍之后，方阵中与点(0，0)相连的数为0点都变成7了，然后遍历方阵的每个边，将 与四个边上数值为0的点 相连的数值为0的连通块变为7。
然后遍历方阵，此时所有的0都在闭合圈内，将其替换为2
最后再遍历方阵，把7换回0。
注：在测试代码时为了方便把每个功能块用不同的函数名封装
注：X[]、Y[]都要为5，第一个数0代表点自身。

#include<bits/stdc++.h>
using namespace std;

int n;
int g[35][35];
int X[5] = {0,-1,0,1,0};
int Y[5] = {0,0,-1,0,1};
bool st[35][35] = {false};

struct node
{
    int x;
    int y;
};

void bfs(int a,int b)
{
    queue<node>q;
    q.push({a,b});

    st[a][b] = true;

    while(!q.empty())
    {
        int dx = q.front().x;
        int dy = q.front().y;
        q.pop();

        for(int i = 0; i < 5; i++)
        {
            int nx = dx + X[i];
            int ny = dy + Y[i];

            if(nx >= 0 && ny >= 0 && nx <= n-1 && ny <= n-1 && g[nx][ny] == 0)
            {
                q.push({nx,ny});
                st[nx][ny] = true;
                g[nx][ny] = 7;
            }
        }
    }
}

void zts()//zero to seven
{
    for(int i = 0; i < n; i++)
    {
        if(g[0][i] == 0)
        {
            bfs(0,i);
        }
        if(g[i][0] == 0)
        {
            bfs(i,0);
        }
        if(g[n-1][i] == 0)
        {
            bfs(n-1,i);
        }
        if(g[i][n-1] == 0)
        {
            bfs(i,n-1);
        }
    }
}

void fun()
{
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < n; j++)
        {
            if(g[i][j] == 7)
            {
                g[i][j] = 0;
            }
        }
    }
}

int main()
{
    cin >> n;
    
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < n; j++)
        {
            cin >> g[i][j];
        }
    }

    bfs(0,0);
    zts();

    
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            if (g[i][j] == 0)
            {
                g[i][j] = 2;
            }
        }
    }

    fun();

    
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < n; j++)
        {
            cout << g[i][j] <<" ";
        }
        cout <<endl;
    }

    return 0;
}