NYOJ—水池数目（DFS）

描述:
校园里有一些小河和一些湖泊，现在，我们把它们通一看成水池，假设有一张我们学校的某处的地图，这个地图上仅标识了此处是否是水池，现在，你的任务来了，请用计算机算出该地图中共有几个水池。
输入:
第一行输入一个整数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来解决此问题，从任意的‘1’开始，不停地把上下左右四个位置的部分用’0’代替。1次DFS后与初始的’1’连接的所有’1’就都被替换成’0’，这样直到图中不存在’1’为止。进行的DFS次数就是水池的数目了。

代码如下：

#include<cstdio>
using namespace std;
const int maxn = 100+5;
int p[maxn][maxn];
int r,c;
void dfs(int i,int j)
{
	if(i-1>=0&&p[i-1][j]==1)
	{
		p[i-1][j]=0;
		dfs(i-1,j);
	}
	if(i+1<r&&p[i+1][j]==1)
	{
		p[i+1][j]=0;
		dfs(i+1,j);
	}
	if(j-1>=0&&p[i][j-1]==1)
	{
		p[i][j-1]=0;
		dfs(i,j-1);
	}
	if(j+1<c&&p[i][j+1]==1)
	{
		p[i][j+1]=0;
		dfs(i,j+1);
	}
	return;
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&r,&c);
		for(int i=0;i<r;i++)
		{
			for(int j=0;j<c;j++)
			{
				scanf("%d",&p[i][j]);
			}
		}
		int k=0;
		for(int i=0;i<r;i++)
		{
			for(int j=0;j<c;j++)
			{
				if(p[i][j]==1)// 每个点都确认一次是否为1
				{
					dfs(i,j);
					k++;
				}
			}
		}
		printf("%d\n",k);
	}
	return 0;
 } 

我这里写麻烦了，应该将上下左右放在一个数组里进行循环的

int dir[4][2]={{-1,0},{1,0},{0,-1},{0,1}};//表示上下左右四个方向