【搜索】洛谷 P1162 填涂颜色

题目描述

由数字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<iostream>
using namespace std;
int a[35][35],b[35][35],n;
int dr[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
void search(int x,int y)
{
    a[x][y]=0;
    for(int i=0;i<4;i++)
    {
        int nx=x+dr[i][0],ny=y+dr[i][1];
        if(a[nx][ny]==0)continue;
        else search(nx,ny);
    }
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cin>>b[i][j];
            if(b[i][j]==1)a[i][j]=0;
            else a[i][j]=1;
        }
    }
    search(1,1);
    search(1,n);
    search(n,1);
    search(n,n);
    for(int i=1;i<=n;i++)
    {
        if(a[i][1]==1)search(i,1);
        if(a[i][n]==1)search(i,n);
        if(a[1][i]==1)search(1,i);
        if(a[n][i]==1)search(n,i);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(a[i][j]==1)b[i][j]=2;
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n-1;j++)cout<<b[i][j]<<' ';
        cout<<b[i][n]<<endl;
    }
    return 0;
}