【基础】填涂颜色

题目描述

由数字0组成的方阵中，有一任意形状闭合圈，闭合圈由数字1构成，围圈时只走上下左右4个方向。现要求把闭合圈内的所有空间都填写成2.例如：6×6 的方阵（n=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≤n≤30)
接下来n行，由0和1组成的n×n 的方阵。
方阵内只有一个闭合圈，圈内至少有一个0。

输出

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

代码

#include <bits/stdc++.h>
using namespace std;
int N;
int fx[5]={0,0,1,0,-1};
int fy[5]={0,1,0,-1,0};
int a[50][50];
void dfs(int x,int y)
{
        for(int i=1;i<5;i++)
        {
            int tx,ty;
            tx=x+fx[i];
            ty=y+fy[i];
            if(x>=1&&y>=1&&a[tx][ty]==0&&x<=N&&y<=N)
            {
                a[tx][ty]=-1;
                dfs(tx,ty);
            }
        }
}
int main()
{
    cin>>N;
    for(int i=1;i<=N;i++)
    {
        for(int j=1;j<=N;j++)
            cin>>a[i][j];
    }
    for(int i=1;i<=N;i++)
    {
        if(a[i][1]!=1)
        {
            a[i][1]=-1;
            dfs(i,1);
        }
  
        if(a[i][N]!=1)
        {
            a[i][N]=-1;
            dfs(i,N);
        }
  
    }
    for(int i=1;i<=N;i++)
    {
        if(a[1][i]!=1)
        {
            a[1][i]=-1;
            dfs(1,i);
        }
  
        if(a[N][i]!=1)
        {
            a[N][i]=-1;
            dfs(N,i);
        }
  
    }
    for(int i=1;i<=N;i++)
    {
        for(int j=1;j<=N;j++)
        {
            if(a[i][j]==-1)cout<<0<<" ";
            else if(a[i][j]==1)cout<<1<<" ";
            else
                cout<<2<<" ";
        }
        cout<<endl;
    }
}