poj 1161 并査集

参考链接1

参考链接2

并査集。

很多学生，每个学生可能属于一个或者多个社团。一个学生染病，其所在社团都有病。问一号学生得病了，一共有多少人得病。

#include <iostream>
#include <cstring>
#include <string>
#include <list>
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <ctime>
#include <string>
#include <map>
#include <algorithm>
#include <sstream>
#include <stack>
#include <set>
#include <queue>
#include <limits>
using namespace std;

#define MAX 30001
int m, n;
int father[MAX], num[MAX];

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

int findset(int x)
{
	if(x != father[x])
		father[x] = findset(father[x]);
	return father[x];
}

void uunion(int a, int b)
{
	int x = findset(a);
	int y = findset(b);
	if(x != y)
	{
		if(num[x] > num[y])
		{
			father[y] = x;
			num[x] += num[y];
		}
		else
		{
			father[x] = y;
			num[y] += num[x];
		}
	}
}

int main()
{
	while(scanf("%d %d", &n, &m) != EOF && n != 0)
	{
		makeset(n);
		for(int i = 0; i < m; ++i)
		{
			int l, first, b;
			scanf("%d %d", &l, &first);
			for(int j = 1; j < l; ++j)
			{
				scanf("%d", &b);
				uunion(first, b);
			}
		}
		cout << num[findset(0)] << endl;
	}

	return 1;
}