填涂颜色

P1162 填涂颜色

#include <iostream>
#include <cstdio>
using namespace std;
int n;
int x=0,y=0;
int a[35][35],b[35][35];
int xx[4]={1,-1,0,0};
int yy[4]={0,0,1,-1};
struct T
{
	int x,y;
}c[1001];
void bfs(int i,int j)//搜索与边界联通的联通块 
{
	int h=0,t=1;
	int p,q;
	c[1].x=i;
	c[1].y=j;
	b[i][j]=1;
	do{
		h++;
		for(int i=0;i<4;i++)
		{
			p=c[h].x+xx[i];
			q=c[h].y+yy[i];
			if(a[p][q]==1 || b[p][q] || p<0 || p>=n || q<0 || q>=n)
			{
				continue;
			}
			t++;
			c[t].x=p;
			c[t].y=q;
			b[p][q]=1;
		}
	}while(h<t);
}
int main(){
	cin>>n;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			cin>>a[i][j];
		}
	}
	for(int j=0;j<n;j++)//第一行边界 
	{
		if(a[0][j] || b[0][j])
			continue;
		bfs(0,j);
	}
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j+=n-1)//下面几行只枚举头和尾 
		{
			if(a[i][j] || b[i][j])
			{
				continue;
			}
			bfs(i,j);
		}
	}
	for(int j=0;j<n;j++)//最后一行边界 
	{
		if(a[n-1][j] || b[n-1][j])
			continue;
		bfs(n-1,j);
	}
	//一定要从边界bfs 
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			if(a[i][j]==1 || b[i][j])
			{
				printf("%d ",a[i][j]);
			}
			else printf("%d ",2);
		}
		printf("\n"); 
	}
	return 0;
}