天天看點

The Suspects (并查集)

題目大意:多個塊,如果一個塊裡有0,則這個塊裡所有人都是患者,如果裡邊的患者又在其他塊裡,那麼另外一個塊裡的也都是患者,以此類推;

題解:很明顯的并查集,用num[]數組記錄之間的關系,先将num數組初始化為1,之後如果a和b之間有關系,将num[a]加到num[b]上,就相當于知道了多一個人是患者。這樣每個點的記錄數都是不一樣的。

其他都是按模闆,Union函數稍微改一下。

完整代碼:

#include <iostream>
#include <stdlib.h>
using namespace std;
#define MAX1 33000
int N,M;
int pre[MAX1];
int num[MAX1];
void  Init()
{
    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];
    return r;
}
void Union(int x,int y)
{
    int a;int b;
    a=find(x);
    b=find(y);
    if(a!=b)
       {
            pre[a]=b;
            num[b]+=num[a];
       }
}
int main()
{
    int k;
    int a[MAX1];
    while(cin>>N>>M,N||M)
    {
        Init();//剛開始忘了寫這個,是以一直輸出0;
        while(M--)
           {
               cin>>k;
               for(int j=0;j<k;j++)
                cin>>a[j];
               for(int i=0;i<k-1;i++)
                Union(a[i],a[i+1]);
           }
    int k=find(0);
    cout<<num[k]<<endl;
    }
    return 0;
}