poj-1611-The Suspects（水並查集）

題目鏈接：http://poj.org/problem?id=1611

題意：輸入n個人，m個組。初始化0爲疑似病例。輸入m個小組，每組中只要有一個疑似病例，整組人都是疑似病例。相同的成員可以在不同的組。找出一共有多少個疑似病例。

解題思路：同組的同parent，查找，合併集合。最後將出現的每個組員的parent和0的parent相比較，統計便可。

AC代碼：

#include<iostream>
#include<algorithm>
using namespace std;
#define ma 300010

int fa[ma],m,n;
int h[ma];
int member[ma];
int Count = 1;
void init(int a)
{
  for(int i = 0; i < a; i++)
  {
    fa[i] = i;
    h[i] = 0;
  }
  return;
}

int Find(int x)
{
  if(fa[x] == x)
   return x;
  else
   return fa[x] = Find(fa[x]);
}

void unite(int x,int y)
{
  x = Find(fa[x]);
  y = Find(fa[y]);

  if(x == y) return;
  else
  {
    if(h[x] > h[y]) fa[y] = fa[x];
    else
    {
      fa[x] = fa[y];
      if(h[x] == h[y]) h[y]++;
    }
  }
}
bool same(int x, int y)
{
 return Find(x) == Find(y);
}

inline void solve(int n,int m)
{
    init(n);
    while(m--)
    {
      int a;
      cin>>a;
      for(int i = 0; i < a; i++)
        cin>>member[i];
      //sort(member,member+a*sizeof(int));
      for(int i = 1; i < a; i++)
       unite(member[i-1],member[i]);
    }
    for(int i = 1; i < n; i++)
      if(same(0,i)) Count++;
    cout<<Count<<endl;

}

int main()
{
  while(cin>>n>>m)
  {
    if(!n) break;
    solve(n,m);
    Count = 1;
  }

  return 0;
}