水池數目(DFS)
阿新 • • 發佈:2019-01-28
水池數目
時間限制: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; }