洛谷 P1162

https://www.luogu.org/problemnew/show/P1162

题目描述

由数字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。

//感谢黄小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

思路：dfs、bfs都行吧，有个转换挺巧妙的。就是从边界开始向内dfs，记录所有0。这些必定是不满足题意的，剩下的肯定就在闭合圈内了。

#include<iostream>
using namespace std;

int a[35][35];
int vis[35][35];
int n;

void dfs(int x,int y);

int main()
{
    cin>>n;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        {
            cin>>a[i][j];
            if(a[i][j])
                vis[i][j]=1;
        }
    for(int i=0;i<n;i++)    //从边缘向内深度搜索 所有0全部变为1
    {
        dfs(i,0);
        dfs(i,n-1);
    }
    for(int i=0;i<n;i++)
    {
        dfs(0,i);
        dfs(n-1,i);
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            if(vis[i][j])
                cout<<"1 ";
            else if(a[i][j])
                cout<<"0 ";
            else
                cout<<"2 ";
        }
        cout<<endl;
    }
    return 0;
}

void dfs(int x,int y)
{
    if(x<0||y<0||x>=n||y>=n||a[x][y]==1)
        return ;
    a[x][y]=1;
    dfs(x+1,y);
    dfs(x,y+1);
    dfs(x-1,y);
    dfs(x,y-1);
}