題目描述:
你有一張某海域NxN像素的照片,".“表示海洋、”#"表示陸地,如下所示:
. . . . . . .
. # # . . . .
. # # . . . .
. . . . # # .
. . # # # # .
. . . # # # .
. . . . . . .
其中"上下左右"四個方向上連在一起的一片陸地組成一座島嶼。例如上圖就有2座島嶼。
由於全球變暖導致了海面上升,科學家預測未來幾十年,島嶼邊緣一個像素的範圍會被海水淹沒。具體來說如果一塊陸地像素與海洋相鄰(上下左右四個相鄰像素中有海洋),它就會被淹沒。
例如上圖中的海域未來會變成如下樣子:
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . # . .
. . . . . . .
. . . . . . .
請你計算:依照科學家的預測,照片中有多少島嶼會被完全淹沒。
思路分析:原先以爲就是典型的利用DFS求解連通塊個數的問題,先計算出原來的連通塊個數,即島嶼數,再計算出淹沒之後的島嶼數,二者相減就行。後來發現數據規模太大了,會大大增加運算時間。這是原來的代碼:
#include<bits/stdc++.h>
using namespace std;
char a[1005][1005],b[1005][1005];
int n,ans1,ans2;
bool vis[1005][1005]={false};
int dir[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
bool in(int x,int y) { //判斷是否越界
return x>=0&&x<n&&y>=0&&y<n;
}
void dfs(int x,int y) {
vis[x][y]=true; //已經訪問
for(int i=0;i<4;i++) {
int dx=x+dir[i][0];
int dy=y+dir[i][1];
if(in(dx,dy)&&!vis[dx][dy]&&a[dx][dy]=='#') {
dfs(dx,dy);
}
}
}
int main() {
cin>>n;
for(int i=0;i<n;i++) {
scanf("%s",a[i]);
}
memcpy(b,a,sizeof(a));
for(int i=0;i<n;i++) {
for(int j=0;j<n;j++) {
if(a[i][j]=='#'&&!vis[i][j]) {
ans1++;
dfs(i,j); //計算原先島嶼數
}
}
}
memset(vis,false,sizeof(vis)); //重新置零
for(int i=0;i<n;i++) {
for(int j=0;j<n;j++) {
if(b[i][j]=='#'&&(in(i-1,j)&&b[i-1][j]=='.'||in(i+1,j)&&b[i+1][j]=='.'||in(i,j-1)&&b[i][j-1]=='.'||in(i,j+1)&&b[i][j+1]=='.')) {
a[i][j]='.'; //島嶼像素被淹沒
}
}
}
for(int i=0;i<n;i++) {
for(int j=0;j<n;j++) {
if(a[i][j]=='#'&&!vis[i][j]) {
ans2++;
dfs(i,j); //計算原先島嶼數
}
}
}
cout<<ans1-ans2;
return 0;
}
BFS也可以求連通塊,不過還不太熟悉,後面慢慢補上。