1. 程式人生 > >nyoj 題目27 水池數目

nyoj 題目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;
}