1. 程式人生 > >水池數目(DFS)

水池數目(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

本題是簡單的搜尋問題,採用深度優先遍歷可以解決,根據題目要求,假設從任意一點值為'1'的出發,將這點的座標上下左右全部用'0'替換,1次DFS後與初始動這個'1'連線的'1'全部被替換成'0',因此,直到圖中不再存在'1'為至,總共進行的DFS的次數就是最後的結果咯!那麼,根據題目要求,有4個方向,時間複雜度為O(4*n*m)。

這題已經AC,下面是具體的程式碼:

#include <iostream>

using namespace std;

char pos[100][100];
int n,m;
void dfs(int x, int y)
{
    if(0<=x&&x<n&&0<=y&&y<m&&pos[x][y]=='1')
    {
        pos[x][y] = '0';
        dfs(x-1,y);
        dfs(x+1,y);
        dfs(x,y-1);
        dfs(x,y+1);
    }
    return ;
}
int main(void)
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n>>m;
        for(int i = 0; i < n; i++)
        for(int j = 0; j < m; j++)
        cin>>pos[i][j];
        int count = 0;
        for(int i = 0;i < n; i++)
        for(int j = 0;j < m; j++)
        {
            if(pos[i][j] == '1')
            {
                dfs(i,j);
                count++;
            }
        }
        cout<<count<<endl;
    }
    return 0;
}