小浣熊鬆鬆和朋友到野外露營,沒想到遇上了π年一次的大洪水,好在鬆鬆是一隻愛觀察的小浣熊,他發現露營地的地形和洪水有如下性質:
①露營地可以被看做是一個N*M的矩形方陣,其中左上角座標爲(1,1),右下角座標爲(n,m),每個格子(i,j)都有一個高度h(i,j)。
②洪水送(r,c)開始,如果一個格子被洪水淹沒,那這個格子四周比它低(或相同)的格子也會被淹沒。
現在鬆鬆想請你幫忙算算,有多少個格子不會被淹沒,便於他和朋友逃脫。
【原有誤數據已刪除】
第一行包含兩個整數n,m,表示矩形方陣右下角座標。
以下n行,每行m個數,第i行第j個數表示格子(i,j)的高度。
最後一行包含兩個整數r,c,表示最初被洪水淹沒的格子。
輸出僅一行,爲永遠不會被淹沒的格子的數量。
3 3
1 2 3
2 3 4
3 4 5
2 2
5
對於90%的數據,保證隨機生成。
對於100%的數據,1<=N,M<=1000。
分類標籤 Tags
深度優先搜索
代碼
#include<cstdio>
using namespace std;
int n0, m0, r, c, cnt, h[1005][1005],yan[1005][1005];
//方陣大小 最先被淹點 剩餘格子數 高度 是否被淹
void dfs(int r,int c,int n,int m)
{ // 現在在第n行m列,水從(r,c)來
if(yan[n][m]==1)return; //已被淹,擴展過
if(m<=0||m>m0||n<=0||n>n0)return;//出界
if(h[n][m]<=h[r][c]) //可被淹
{
yan[n][m]=1;
dfs(n,m,n+1,m);
dfs(n,m,n-1,m);
dfs(n,m,n,m+1);
dfs(n,m,n,m-1);
}
}
int main()
{
scanf("%d%d",&n0,&m0);
for(int i=1;i<=n0;i++)
for(int j=1;j<=m0;j++)
scanf("%d",&h[i][j]);
scanf("%d%d",&r,&c);
yan[r][c]=1; //DFS
dfs(r,c,r+1,c);
dfs(r,c,r-1,c);
dfs(r,c,r,c+1);
dfs(r,c,r,c-1);
for(int i=1;i<=n0;i++)
for(int j=1;j<=m0;j++)
if(yan[i][j]==0)cnt++;
printf("%d\n",cnt);
return 0;
}