洛谷 P1162 填涂颜色题解

题目描述

由数字00组成的方阵中，有一任意形状闭合圈，闭合圈由数字11构成，围圈时只走上下左右44个方向。现要求把闭合圈内的所有空间都填写成22.例如：6 \times 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 \times 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 = ;
 int n, map[MAXN][MAXN], vis[MAXN][MAXN];
 int pos[][] = {, , , -, , , -, };

 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] != )
     {
         return;
     }
     front = rear = ;
     q[rear] = now;
     rear++;
     while(front < rear)
     {
         now = q[front++];
         if(map[now.x][now.y] == )
         {
             map[now.x][now.y] = ;
         }
         if(now.x == )
         {
             now.x = ;
         }
         for(int i = ; i < ; i++)
         {
             int nx = now.x + pos[i][];
             int ny = now.y + pos[i][];
             if(nx <= n && nx >  && ny <= n && ny >
                 && vis[nx][ny] == false
                 && map[nx][ny] == )
             {
                 map[nx][ny] = ;
                 vis[nx][ny] = true;
                 q[rear].x = nx;
                 q[rear].y = ny;
                 rear++;
             }
         }
     }
 }

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

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