洛谷 P1162 填涂颜色题解

题目描述

由数字00组成的方阵中，有一任意形状闭合圈，闭合圈由数字11构成，围圈时只走上下左右44个方向。现要求把闭合圈内的所有空间都填写成22.例如：6 imes 66×6的方阵（n=6n=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

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

输入格式

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

接下来nn行，由00和11组成的n imes nn×n的方阵。

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

//感谢黄小U饮品指出本题数据和数据格式不一样. 已修改(输入格式)

输出格式

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

输入输出样例

输入 #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

说明/提示

1 le n le 301≤n≤30

---

题解

这道题有一个简单的算法就是输入时将所有为0的数据都填写为2，然后从4个边向内部进行BFS，如果搜索到2就将其改为0，并继续搜索，如果搜索到1或0就停止搜索。

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <algorithm>
#include <string.h>

using namespace std;

const int MAXN = 105;
int n, map[MAXN][MAXN], vis[MAXN][MAXN];
int pos[4][2] = {0, 1, 0, -1, 1, 0, -1, 0};

struct Node
{
    int x, y;
};

Node q[MAXN];
int front, rear;
int a, b;

void bfs()
{
    Node now;
    now.x = a;
    now.y = b;
    if(map[a][b] != 2)
    {
        return;
    }
    front = rear = 0;
    q[rear] = now;
    rear++; 
    while(front < rear)
    {
        now = q[front++]; 
        if(map[now.x][now.y] == 2)
        {
            map[now.x][now.y] = 0;
        }
        if(now.x == 7)
        {
            now.x = 7;
        }
        for(int i = 0; i < 4; i++) 
        {
            int nx = now.x + pos[i][0]; 
            int ny = now.y + pos[i][1]; 
            if(nx <= n && nx > 0 && ny <= n && ny > 0  
                && vis[nx][ny] == false 
                && map[nx][ny] == 2) 
            {
                map[nx][ny] = 0;
                vis[nx][ny] = true;
                q[rear].x = nx;
                q[rear].y = ny;
                rear++;
            }
        } 
    }
}


int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= n; j++)
        {
            cin >> map[i][j];
            if(map[i][j] == 0)
            {
                map[i][j] = 2;
            }
        }
    }
    for(int i = 1; i <= n; i++)
    {
        a = 1;
        b = i;
        bfs();
    }
    for(int i = 1; i <= n; i++)
    {
        a = n;
        b = i;
        bfs();
    }
    for(int i = 1; i <= n; i++)
    {
        a = i;
        b = 1;
        bfs();
    }
    for(int i = 1; i <= n; i++)
    {
        a = i;
        b = n;
        bfs();
    }
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= n; j++)
        {
            cout << map[i][j] << " ";
        }
        cout << endl;
    }
    return 0;
}

这个BFS并不难写，不过当时犯了一个小错误，导致2/3/4个样例都是WA，特别是第2个样例在本机输出的结果和标准答案一致，但是提交后总是说一个位置应该为0，输出了2。查了很久，最后发现是把pos[4][2]写成了pos[2][4]。数组定义错了，导致遍历时移动的位置错误了，而且本机的对应内存的数据和测试机不同，所以在本机上是过了，但是测试机没有过。