洛谷—— P1162 填涂颜色

https://www.luogu.org/problem/show?pid=1162

题目描述

由数字0 组成的方阵中，有一任意形状闭合圈，闭合圈由数字1构成，围圈时只走上下左右4个方向。现要求把闭合圈内的所有空间都填写成2.例如：6X6的方阵（n=6），涂色前和涂色后的方阵如下：

0 0 0 0 0 0 0 0 0 0 0 0

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

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

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

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

1 1 1 1 1 1 1 1 1 1 1 1

输入输出格式

输入格式：

每组测试数据第一行一个整数：n。其中n(1<=n<=30)

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

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

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

输出格式：

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

输入输出样例

输入样例#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<=n<=30

#include <algorithm>
#include <cstdio>

using namespace std;

int n,num[31][31];
bool mark[31][31];
int fx[4]={0,0,1,-1};
int fy[4]={1,-1,0,0};

void BFS(int x,int y)
{
    mark[x][y]=1;
    for(int i=0;i<4;i++)
    {
        int xx=x+fx[i],yy=y+fy[i];
        if(mark[xx][yy]||num[xx][yy]==1) continue;
        if(xx<0||yy<0||xx>n||yy>n) continue;
        BFS(xx,yy);
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
      for(int j=1;j<=n;j++)
          scanf("%d",&num[i][j]);
    for(int i=1;i<=n;i++)
    {
        if(num[1][i]==0) BFS(1,i);
        if(num[n][i]==0) BFS(n,i); 
        if(num[i][1]==0) BFS(i,1);
        if(num[i][n]==0) BFS(i,n);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
            if(mark[i][j]&&num[i][j]==0) printf("%d ",0);
            else if(num[i][j]==0) printf("%d ",2);
            else printf("%d ",num[i][j]);
        printf("
");
    }
    return 0;
}