1. 程式人生 > >POJ 1611 -- The Suspects (並查集)

POJ 1611 -- The Suspects (並查集)

題意

在一所大學裡有n(0 < n <= 30000)個人,並有m(0 <= m <= 500)個團體組織 組織人數不等。這n個人從0 ~ n-1 編號,其中0號是感染SARS的嫌疑者,被判定為有嫌疑的機制是與嫌疑者在同一組織內。問共有多少嫌疑者?

思路

運用並查集的方法,合併與查詢,在輸入的過程中把一個個分開的集合合併成一個個大的集合,最後查詢0號。

合併:構造陣列,為n個學生的各自的集合,合併的過程中集合與集合之間形成一棵棵的樹,每個節點的根節點是各自的上一層的              節點,樹的根節點是它本身。

查詢:從當前節點一層層向上查詢根節點的過程。

code

#include<stdio.h>
#include<string.h>

int pre[50005];
int num[50005];

void init(int n)
{
    for(int i=0;i<n;i++)
    {
        pre[i]=i;
        num[i]=1;
    }
}

int Find(int x)
{
    int r=x;

    while(pre[r]!=r)
        r=pre[r];
/*
    int i=x,j;  //有無壓縮路徑的情況

    while(i!=r)
    {
        j=pre[i];
        pre[i]=r;
        i=j;
    }
*/
    return r;
}

void join(int x,int y)
{
    int fx=Find(x);
    int fy=Find(y);

    if(fx!=fy)
    {
        pre[fx]=fy;
        num[fy]+=num[fx];
    }
}

int main()
{
    int n,m,k,x,y;

    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==0 && m==0)
            break;

        init(n);

        for(int i=0;i<m;i++)
        {
            scanf("%d",&k);
            scanf("%d",&x);

            for(int i=1;i<k;i++)
            {
                scanf("%d",&y);
                join(x,y);
                x=y;
            }
        }


        int p=Find(0);
        printf("%d\n",num[p]);

        for(int i=0;i<n;i++)
            printf("%d ",pre[i]);
    }
    return 0;
}
/*無壓縮路徑*/
/*
100 4
2 1 2
5 1 10 13 11 14
2 0 1
2 99 2
8
14 2 10 3 4 5 6 7 8 9 13 14 12 11 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 14
*/
/*有壓縮路徑*/
/*
100 4
2 1 2
5 1 10 13 11 14
2 0 1
2 99 2
8
14 14 14 3 4 5 6 7 8 9 14 14 12 14 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 14
*/

不可不看這篇部落格,很有意思。

https://blog.csdn.net/u013546077/article/details/64509038