P1162 填涂颜色

本题为洛谷

这道题用了广搜 ：

#include<iostream>
#include<queue>
using namespace std;
int n,a[31][31],dir[4][2]={{1,0},{0,1},{0,-1},{-1,0}};
struct point{
	int x,y;
	point(int xx,int yy){
		x=xx;
		y=yy;
	}
};
void bfs(int sx,int sy){
	queue<point> q;
	q.push(point(sx,sy));
	a[sx][sy]=0;
	while(!q.empty()){
		point p=q.front();
		for(int i=0;i<4;i++){
			int dx=p.x+dir[i][0];
			int dy=p.y+dir[i][1];
			if(dx>0&&dx<=n&&dy>0&&dy<=n&&a[dx][dy]==2){
				q.push(point(dx,dy));
				a[dx][dy]=0;
			}
		}
		q.pop();
	}
} 
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>a[i][j];
			if(a[i][j]==0) a[i][j]=2;
		}
	}
	for(int i=1;i<=n;i++){
		if(a[i][1]==2) bfs(i,1);
		if(a[i][n]==2) bfs(i,n);
	}
	for(int j=1;j<=n;j++){
		if(a[1][j]==2) bfs(1,j);
		if(a[n][j]==2) bfs(n,j);
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cout<<a[i][j]<<" ";
		}
		cout<<endl;
	}
	return 0;
}