nyoj 题目27 水池数目



http://acm.nyist.net/JudgeOnline/problem.php?pid=27

水池数目

时间限制： 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和0的意义弄反了，w了好几次，以后审题要认真

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define MAX 10000
//把1和0的意义搞反了，w了好几次
struct SET
{
    int S;//按照顺序是第几个数，从0开始
    int x;//相应位置的所在行
    int y;//相应位置的所在列
    int J;//存放1和0
}Set[MAX+10];
int Find(int x)
{
    int r=x,i=x,j;
    while(Set[r].S!=r)
    r=Set[r].S;

    while(i!=r)
    {
        j=Set[i].S;
        Set[i].S=r;
        i=j;
    }
    return r;
}
void Merge(int x,int y)
{
    x=Find(x);
    y=Find(y);
    if(y<x)
    Set[x].S=y;
    if(x<y)
    Set[y].S=x;
}
int main()
{
    int i,j,k,T,M,N,sum,num;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d %d",&M,&N);
        for(i=0;i<M;i++)//给Set[].S赋值0~M*N-1
            for(j=0;j<N;j++)
            {
                num=N*i+j;
                scanf("%d",&Set[num].J);
                Set[num].S=num;
                Set[num].x=i;
                Set[num].y=j;
            }
        for(i=0;i<M;i++)
            for(j=0;j<N;j++)
           {//每一次都执行下边四个if，保证指向最小的数(通过并查集把构成一个湖的指向最小的那个数)
                num=N*i+j;
                //下面分别搜索上下左右，如果是0的话就合并
                if(Set[num].J==1&&i-1>=0&&Set[N*(i-1)+j].J==1)//上
                Merge(num,N*(i-1)+j);

                if(Set[num].J==1&&i+1<M&&Set[N*(i+1)+j].J==1)//下
                Merge(num,N*(i+1)+j);

                if(Set[num].J==1&&j-1>=0&&Set[N*i+j-1].J==1)//左
                Merge(num,N*i+j-1);

                if(Set[num].J==1&&j+1<N&&Set[N*i+j+1].J==1)//右
                Merge(num,N*i+j+1);
           }
 //       for(i=0;i<M*N;i++)
//        {
//            printf("%d#%d ",Set[i].S,i);
//            if((i+1)%N==0)
//                printf("\n");
//        }
        for(i=0,sum=0;i<M*N;i++)//扫描有几个集合，即有几个湖
        if(Set[i].S==i&&Set[i].J==1)
        sum+=1;

        printf("%d\n",sum);
    }
    return 0;
}