1. 程式人生 > >NYOJ 27 水池數目 【DFS】

NYOJ 27 水池數目 【DFS】

水池數目

時間限制:3000 ms  |  記憶體限制:65535 KB 難度:4
描述
南陽理工學院校園裡有一些小河和一些湖泊,現在,我們把它們通一看成水池,假設有一張我們學校的某處的地圖,這個地圖上僅標識了此處是否是水池,現在,你的任務來了,請用計算機算出該地圖中共有幾個水池。
輸入
第一行輸入一個整數N,表示共有N組測試資料
每一組資料都是先輸入該地圖的行數m(0<m<100)與列數n(0<n<100),然後,輸入接下來的m行每行輸入n個數,表示此處有水還是沒水(1表示此處是水池,0表示此處是地面)
輸出
輸出該地圖中水池的個數。
要注意,每個水池的旁邊(上下左右四個位置)如果還是水池的話的話,它們可以看做是同一個水池。
樣例輸入
2
3 4
1 0 0 0 
0 0 1 1
1 1 1 0
5 5
1 1 1 1 0
0 0 1 0 1
0 0 0 0 0
1 1 1 0 0
0 0 1 1 1
樣例輸出
2
3

/*題解: 
    對每一步進行判斷,如果是水池並且尚未標記則進行DFS搜尋,對搜尋過的位置坐下標記。
最後,進行DFS搜尋的次數,即是水池的個數。 
*/

#include<cstdio>
#include<cstring>
int map[110][110],dir[4][2]={0,1,0,-1,1,0,-1,0},vis[110][110];
int a,b,m,n;
void dfs(int x,int y)
{
    if(!map[x][y]) return;//結束條件,map[x][y]不為水池。
    for(int i=0; i<4; i++)//向上、下、左、右搜尋
    {
        a=x+dir[i][0];
        b=y+dir[i][1];
        if(a>=0&&b>=0&&a<m&&b<n&&vis[a][b]==0&&map[a][b])
        {
            vis[a][b]=1;
            dfs(a,b);
        }
    }
} 
int main(){
     int N,i,j,count;
     scanf("%d",&N);
     while(N--)
     {
        scanf("%d %d",&m,&n);
        for(i=0; i<m; i++)
          for(j=0; j<n; j++)
            scanf("%d",&map[i][j]);
        memset(vis,0,sizeof(vis));
        for(i=0,count=0; i<m; i++)
        {
            for(j=0; j<n; j++)
            {
                if(!vis[i][j]&&map[i][j])//未標記且為水池
                {
                    dfs(i,j);
                    count++;
                }
            }
        }
        printf("%d\n",count);
    }
    return 0;
}