dfs 填充颜色

题目描述

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

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

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

输入输出格式

输入格式：

每组测试数据第一行一个整数n(1 \le n \le 30)n(1≤n≤30)

接下来nn行，由00和11组成的n \times nn×n的方阵。

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

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

输出格式：

扫描二维码关注公众号，回复： 4239673 查看本文章

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

输入输出样例

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

分析：可以从四周开始dfs能搜到的标记一下，那么搜不到的就是中间被包围的。

代码如下：

#include<iostream>
#include<cstdio>
using namespace std;
int a[100][100],b[100][100]={0};
int n,x,y;

void dfs(int x,int y)
{
    if(x>n||x<1||y>n||y<1||a[x][y]!=0)//判断是否越界 
        return;
    a[x][y]=1;
    dfs(x+1,y);
    dfs(x-1,y);
    dfs(x,y+1);
    dfs(x,y-1);//四个方向 
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            cin>>a[i][j];
            if(a[i][j]==1) b[i][j]=-1;//b用来存原来的位置 
        }
    int flag=0;
    for(int i=1;i<=n;i++)
    {
    if(a[i][1]!=1) dfs(i,1);
    if(a[i][n]!=1) dfs(i,n);//搜第i行的第一列和第n列 
    }
    for(int i=1;i<=n;i++)
    {
    if(a[1][i]!=1) dfs(1,i);
    if(a[n][i]!=1) dfs(n,i);//搜第i列的第一行和第n行 
    } 
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(b[i][j]==-1) printf("1 ");
            else
            if(a[i][j]==0) printf("2 ");
            else printf("0 ");
        }
        printf("\n");
    }//输出，如果这个看不懂的话建议刷P1000(大牛分站的！); 
    return 0;
}
//代码如下：