【搜索】洛谷 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;}