NYOJ - 27 水池数目

             描述

校园里有一些小河和一些湖泊，现在，我们把它们通一看成水池，假设有一张我们学校的某处的地图，这个地图上仅标识了此处是否是水池，现在，你的任务来了，请用计算机算出该地图中共有几个水池。

输入

第一行输入一个整数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

 

 

很简单的一个宽搜问题，记得刚开始ACM的时候，这样的题得做一晚上，然而现在十几分钟一遍就AC了，要继续努力。

 

#include <cstdio>
#include <cstring>

int maps[105][105];
int vis[105][105];
int step[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
int m, n;
int ans;

void solve(int x, int y);
int canput(int x, int y);

int main()
{
    int T;
    scanf("%d", &T);
    while (T--)
    {
        scanf("%d%d", &m, &n);
        for (int i = 0; i < m; ++i)
            for (int j = 0; j < n; ++j)
                scanf("%d", &maps[i][j]);
        ans = 0;
        for (int i = 0; i < m; ++i)
        {
            for (int j = 0; j < n; ++j)
            {
                if (canput(i, j) == 1)
                {
                    solve(i, j);
                    ans++;
                }
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}

int canput(int x, int y)
{
    if ((x < 0 || y < 0 || x >= m || y >= n) || (maps[x][y] == 0))
        return 0;
    return 1;
}

void solve(int x, int y)
{
    if (maps[x][y] == 1)
        maps[x][y] = 0;
    for (int i = 0; i < 4; ++i)
    {
        if (canput(x+step[i][0], y+step[i][1]) == 1)
        {
            solve(x+step[i][0], y+step[i][1]);
        }
    }
}