水池数目

水池数目

时间限制：3000 ms  |  内存限制：65535 KB

难度：4

描述

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

输入

第一行输入一个整数N，表示共有N组测试数据
每一组数据都是先输入该地图的行数m(0<m<100)与列数n(0<n<100)，然后，输入接下来的m行每行输入n个数，表示此处有水还是没水（1表示此处是水池，0表示此处是地面）

输出

输出该地图中水池的个数。
要注意，每个水池的旁边（上下左右四个位置）如果还是水池的话的话，它们可以看做是同一个水池。

样例输入

23 41 0 0 0 0 0 1 11 1 1 05 51 1 1 1 00 0 1 0 10 0 0 0 01 1 1 0 00 0 1 1 1

样例输出

23

本题用深度优先搜索的方法对每个cell元素进行搜索。搜索前提为cell.value的值为1并且cell.visited为false,表示该cell没有被访问过。将访问过的cell元素visited置为true。

本题源码:

#include <iostream>#include <string.h>using namespace std;typedef struct cell{    int value;    bool visited;  //是否被访问}Cell;Cell m_cell[102][102]; //周围元素置0，相当增加一圈土地void Search(int x,int y){ //深度优先搜索    int dir = 4;    while(dir>0){        switch(dir){            case 4://向上访问                if(m_cell[x-1][y].value==1&&                        !m_cell[x-1][y].visited){                    m_cell[x-1][y].visited = true;                    Search(x-1,y);                }                break;            case 3://向下访问                if(m_cell[x+1][y].value==1&&                        !m_cell[x+1][y].visited){                    m_cell[x+1][y].visited = true;                    Search(x+1,y);                }                break;            case 2://向左访问                 if(m_cell[x][y-1].value==1&&                         !m_cell[x][y-1].visited){                    m_cell[x][y-1].visited = true;                    Search(x,y-1);                }                break;            case 1://向右访问                if(m_cell[x][y+1].value==1&&                    !m_cell[x][y+1].visited){                    m_cell[x][y+1].visited = true;                    Search(x,y+1);                }                break;        }            dir--;            }}int main(){    int test;    cin>>test;    while(test--){        int m,n,count=0,i,j; //m行,n列        cin>>m>>n;        memset(m_cell,0,sizeof(m_cell));        for(i=1;i<=m;i++){            for(j=1;j<=n;j++){                cin>>m_cell[i][j].value;            }        }        for(i=1;i<=m;i++){            for(j=1;j<=n;j++){                if(m_cell[i][j].value==1&&!m_cell[i][j].visited){                    //如果当前元素值为1，且没有被访问过则进行遍历                    m_cell[i][j].visited = true;                    Search(i,j);                    count++;                }            }        }        cout<<count<<endl;    }    return 0;}

对上面的代码进行优化，不设置结构体，直接设置一个数组，将深度优先搜索到的元素值直接赋值为0

#include <iostream>#include <string.h>using namespace std;int cell[102][102]; void Search(int x,int y){ //深度优先搜索 int dir = 4;cell[x][y] = 0;while(dir>0){switch(dir){case 4://向上访问 if(cell[x-1][y]==1){Search(x-1,y);}break;case 3://向下访问if(cell[x+1][y]==1){ Search(x+1,y);} break; case 2://向左访问 if(cell[x][y-1]==1){ Search(x,y-1);} break; case 1://向右访问 if(cell[x][y+1]==1){ Search(x,y+1);} break;}dir--;}}int main(){int test;cin>>test;while(test--){int m,n,count=0,i,j; //m行,n列 cin>>m>>n; memset(cell,0,sizeof(cell));for(i=1;i<=m;i++){for(j=1;j<=n;j++){cin>>cell[i][j];}}for(i=1;i<=m;i++){for(j=1;j<=n;j++){if(cell[i][j]==1){Search(i,j);count++;}}}cout<<count<<endl;}return 0;}