1. 程式人生 > >洛谷 P1162 填充顏色 (BFS)

洛谷 P1162 填充顏色 (BFS)

圖片 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)