1. 程式人生 > >nyoj ACM:水池數目(dfs)

nyoj ACM:水池數目(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. 就是遇到一個點,壓棧
2. 看他下一個點可不可以訪問,如果可以,重複第1步
3. 如果不可以,換一個方向,重複第2步。直到4個方向都考慮到了,就捨棄他,pop()掉

每個節點訪問完後,有4個發現可以走,上下左右。用visited[i][j]表示第i行第j列出發的次數,visited[i][j]初始化為0,每以改點出發一次,visited[i][j]++ 。出發的發現 就是 visited[i][j]的值(1-右,2-下,3-左,4-上)。所以從next出發訪問下一個點fr的座標就是
visited[next.x][next.y]++;
fr.x=next.x+d [visited[next.x][next.y]] [0];
fr.y=next.y+d [visited[next.x][next.y]] [1];

如果visited[next.x][next.y]大於4時,表示節點從next出發的上下左右4個方向已經都走遍了,這個點捨棄掉,pop().


#include<iostream>
#include<cstdio>
#include<cstring>
#include<stack>
using namespace std;
typedef struct{
    int x;
    int y;
    int c;
}node;
int a[100][100];
int visited[100][100];

int d[5][2]={{0,0},{0,1},{1
,0},{0,-1},{-1,0}}; void dfs(int m,int n) { int p,q; int num=0; stack<node> mystack; node fr,next; memset(visited,0,sizeof(visited)); for(int i=0;i<m;i++) { for(int j=0;j<n;j++) { if(a[i][j]==1&&visited[i][j]<4) { next.x=i; next.y=j; mystack.push(next); num++; } while(!mystack.empty()) { next=mystack.top(); if(visited[next.x][next.y]<4) { visited[next.x][next.y]++; fr.x=next.x+d[visited[next.x][next.y]][0]; fr.y=next.y+d[visited[next.x][next.y]][1]; } else { mystack.pop(); continue; } while((fr.x<m && fr.x>=0 && fr.y>=0 && fr.y<n) && a[fr.x][fr.y]==1 && visited[fr.x][fr.y]<4) { mystack.push(fr); visited[fr.x][fr.y]++; next=fr; fr.x=next.x+d[visited[next.x][next.y]][0]; fr.y=next.y+d[visited[next.x][next.y]][1]; } } } } printf("%d\n",num); return; } int main() { int N; int k; int m,n; scanf("%d",&N); for(k=1;k<=N;k++) { scanf("%d%d",&m,&n); for(int i=0;i<m;i++) for(int j=0;j<n;j++) scanf("%d",&a[i][j]); dfs(m,n); } return 0; }