1外面有1圈2

P1162 填涂颜色

a6b6c6d6 @ 2024-12-19 20:55:55

#include<bits/stdc++.h>
using namespace std;
int n,ax,ay;
int a[31][31],b[31][31];
bool flag[31][31];
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
bool ext(int ax,int ay){
    return ax>=1&&ax<=n&&ay>=1&&ay<=n&&a[ax][ay]==2;
}
void bfs(int ax,int ay){
    queue< pair<int,int> >q;
    q.push(make_pair(ax,ay));
    while(!q.empty()){
        int x=q.front().first;
        int y=q.front().second;
        q.pop();
        a[x][y]=0;
        for(int i=0;i<4;i++){
            if(ext(ax+dx[i],ay+dy[i]))q.push(make_pair(ax+dx[i],ay+dy[i]));
        }
    }
}
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++){
        for(int j=1;j<=n;j++){
            if((i==1||i==n)&&(j==1||j==n)&&a[i][j]==2){
                ax=i;
                ay=j;
                break;
            }
        }
    }
    bfs(ax,ay);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cout<<a[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}

|