填涂颜色【洛谷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。

输出格式
已经填好数字2的完整方阵。

思路
关键在于在哪里填2,外面的0搜一遍，剩下的位置都填2
用bfs会快一点
code

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int n,a[1005][1005],vst[1005][1005],flag=1; 
int dx[4]={1,0,-1,0},dy[4]={0,1,0,-1};
struct node 
{
	int x,y;
};
queue<node>q;
int BFS(int x,int y) 
{  
   node t;
   t.x=x;t.y=y;
   q.push(t); 
   flag++;
   vst[x][y]=flag;
   int fl=1; 
   while(!q.empty())
   {  
   		int nx=q.front().x,ny=q.front().y;
		for(int i=0;i<4;i++)
      	{  
		  	int tx=nx+dx[i],ty=ny+dy[i];
         	if(tx<1||tx>n||ty<1||ty>n)
			 {
			 	fl=0;
			 	continue;
			} 
         	if(!vst[tx][ty])
         	{  
			 	node tt;
				tt.x=tx;
				tt.y=ty;
            	q.push(tt);
            	vst[tx][ty]=flag;  
        	}
      }
   	}
   	q.pop();
}
   return fl;
}
int main()
{  
	scanf("%d",&n);
	int i,j,ii,jj;
   	for(i=1;i<=n;i++)
     	for(j=1;j<=n;j++)
     	{  
		 	scanf("%d",&a[i][j]);
        	if(a[i][j])
				vst[i][j]=1; 
     	}
   	for(i=2;i<=n;i++)
     for(j=2;j<=n;j++) 
       if(!vst[i][j])
         if(BFS(i,j)) 
         {  
		 	for(ii=1;ii<=n;ii++)
            {  
				for(jj=1;jj<=n;jj++)
                 	if(vst[ii][jj]==flag)
				 		printf("2 "); 
                 	else 
					 	printf("%d ",a[i][j]);
                printf("\n");
            }
         }
   	return 0;
 }