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