洛谷 P1162 填涂颜色

填涂颜色

题目描述

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

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

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

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

输出格式

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

样例 #1

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

提示

对于 $100\%$ 的数据，$1\le n\le 30$。

分析

搜索

代码

#include<bits/stdc++.h>

using namespace std;

int a[40][40],b[40][40];

int dx[5]={0,-1,1,0,0};

int dy[5]={0,0,0,-1,1};//习惯下标从1开始 

int n; 
void dfs(int p,int q)
{
    if(p<0||p>n+1||q<0||q>n+1||a[p][q]!=0) 
    {
		return;//如果是边界或者已经被搜过了或者本来就是墙的就反回
    }
    
    a[p][q]=1;//标记，染色
    
    for(int i=1;i<=4;i++) 
	{
		dfs(p+dx[i],q+dy[i]);//向四个方向搜索
	} 
}

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]==0) 
            {
				a[i][j]=0;
            }
			else 
			{
				a[i][j]=2;
			}
        }
    }
    
    dfs(0,0);//搜索
    
    for (int i=1;i<=n;i++)
	{
        for (int j=1;j<=n;j++)
        {
        	if(a[i][j]==0) 
			{
				cout<<2<<' ';//没搜到 
        	}
        	else 
        	{ 
				cout<<b[i][j]<<' ';
			} 
        }
        
        cout<<endl;
    }
    
    return 0;
}