洛谷 P1162 填充颜色 （BFS）

#include<iostream>
#include<cstdio>
using namespace std;
int map[31][31];
bool flag[5],vis[31][31]; //flag用来储存0的四周是否都有1 
                          //vis用于广搜，判断这个点是否更改过 
int q[1001][2],h=1,t;     //队列 
int fx[5][2]={{0,0},{1,0},{-1,0},{0,1},{0,-1}};//枚举方向 
int main()
{
    //输入 
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
      for(int j=1;j<=n;j++)
        cin>>map[i][j];
    //枚举每个点，判断它的四周是否都有1 
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
          
          if(!map[i][j])  //如果这个点不是1，那么枚举四周，算是个剪枝 
            for(int k=1;k<=n;k++)
            {
                if(i+k<=n&&map[i+k][j]==1)
                  flag[1]=1;
                if(j+k<=n&&map[i][j+k]==1)
                  flag[2]=1;
                if(i-k>=1&&map[i-k][j]==1)
                  flag[3]=1;
                if(j-k>=1&&map[i][j-k]==1)
                  flag[4]=1;  
            }
            if(flag[1]&&flag[2]&&flag[3]&&flag[4]) //如果四周都有1，那么标记，退出循环 
            {
                q[++t][0]=i, q[t][1]=j;
                break;
            }
            for(int k=1;k<=4;k++) //如果不是，清空flag 
              flag[k]=0;
        }
        if(q[t][0])
          break;
    }
    map[q[t][0]][q[t][1]]=2; //把第一个被找到的点变成2 
    while(h<=t) //广搜 
    {
        for(int i=1;i<=4;i++) //枚举四个方向 
        {
            int x=q[h][0]+fx[i][0];
            int y=q[h][1]+fx[i][1];
            if(x<=n&&x>=1&&y<=n&&y>=1&&map[x][y]!=1&&!vis[x][y]) //如果没有越界，且不是1，且没有被访问过 
            {
                map[x][y]=2;//把它变成2         
                q[++t][0]=x;//入队 
                q[t][1]=y;
                vis[x][y]=1;//标记这个点被访问过 
            }
        }
        ++h; //取出队首 
    }
    //输出 
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
          cout<<map[i][j]<<" ";
        cout<<endl;
    }  
    
    return 0;
}