【C4】【数据结构】方阵上色

题目描述

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

注意：方阵内闭合圈可能有多个，也可能没有 闭合圈。

输出

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

样例输入

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

提示

来源

★★★★

#include<bits/stdc++.h>
using namespace std;
int n,mp[1005][1005],dx[4]={1,-1,0,0},dy[4]={0,0,1,-1};
bool vis[1005][1005];
struct no{int x,y;}a[1005];
void bfs(int x,int y){
    int f=0,r=0;no p={x,y};a[0]=p;vis[x][y]=1;
    while(f<=r){
        no p1=a[f];
        for(int i=0;i<4;i++){
            int xx=p1.x+dx[i],yy=p1.y+dy[i];
            if(xx>=1&&xx<=n&&yy>=1&&yy<=n&&mp[xx][yy]==0&&vis[xx][yy]==0){
                r++;no p2={xx,yy};a[r]=p2;vis[xx][yy]=1;
            }
        }
        f++;}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){cin>>mp[i][j];}
	}
	for(int i=1;i<=n;i++){
		if(mp[1][i]==0&&vis[1][i]==0){bfs(1,i);}
		if(mp[n][i]==0&&vis[n][i]==0){bfs(n,i);}
		if(mp[i][1]==0&&vis[i][1]==0){bfs(i,1);}
		if(mp[i][n]==0&&vis[i][n]==0){bfs(i,n);}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(vis[i][j]==0&mp[i][j]==0){
				cout<<2<<" ";
			}else{
				cout<<mp[i][j]<<" ";}}
		cout<<endl;
	}
	return 0;}