洛谷 P1162 填充顏色 (BFS)
阿新 • • 發佈:2018-03-04
圖片 stream http spa 一個 廣搜 清空 AC tdi
這道題的核心就是找到一個被“1”圍起來的“0”,找到以後這道題就是個水題。
我的方法十分無腦,如果這個“0”被“1”圍起來了,那麽它的四周無論多遠(<n)一定有“1”。所以這道題就可以開心的枚舉了(),最差時間復雜度O(n^3),實際要比這跑得快.
#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;
}
洛谷 P1162 填充顏色 (BFS)