GDUT专题学习3 Learning Languages 并查集

题目链接:专题训练3-图论 - Virtual Judge (vjudge.net)

题目大意:

一个公司有n个员工和m种工作语言,每个员工都会0-m(包含0)种语言，会同一种语言的员工可以互相交流，不会同一种语言的员工也可以通过其他员工间接交流，已知，让一个员工学习一种语言的成本为1块钱,公司为了所有员工能够互相交流，至少需要多少钱。

思路:

①如果公司全部人都不会任何一种语言，那么需要他们全部人学习同一种语言，需要花费n块钱;

②用并查集的思想将能交流的人分成k堆，那么将这k堆连接起来的成本需要k-1。

代码:

#include<iostream>
#include<cstring>
#include<set>
using namespace std;
typedef long long uu;

int a[200][200],ans;
int father[200];

int get(int x)
{
	if (father[x] == x)
		return x;
	else
		return father[x] = get(father[x]);
}
void merge(int x, int y)
{
	x = get(x);
	y = get(y);
	if (x != y)
	{
		father[y] = x;
	}
}


int main()
{
	set<int> f;
	//freopen("input.txt","r",stdin);
	int i,j,k,o,n,m,num,x,l1,l2,flag=0;
	cin>>n>>m;
	for(i=1; i<=n; i++) father[i]=i;
	for(i=1; i<=n; i++)
	{
		cin>>num;
		a[i][0]=num;
		for(j=1; j<=num; j++)
			cin>>a[i][j];
	}
	for(i=2; i<=n; i++)
	{
		l1=a[i][0];
		if(l1==0)
			continue;
		for(j=1; j<i; j++)//遍历i之前的员工
		{
			l2=a[j][0];
			if(l2==0)
				continue;
			for(k=1; k<=l2; k++)
			{
				for(o=1; o<=l1; o++)
				{
					if(a[j][k]==a[i][o])//有共同语言
						merge(i,j);//并在一起
				}
			}
		}
	}
	for(i=1; i<=n; i++)
	{
		f.insert(get(i));
		if(a[i][0]==0) flag++;
	}
	if(flag==n) cout<<n;
	else cout<<f.size()-1;
}