本题为洛谷
这道题用了广搜 :
#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;
}