Cod sursa(job #573680)

Utilizator mihai995mihai995 mihai995 Data 6 aprilie 2011 14:42:54
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.73 kb
#include <fstream>
using namespace std;

const int N=100005;
const char rasp[][6]={"NU\n","DA\n"};
int v[N],size[N];

ifstream in("disjoint.in");
ofstream out("disjoint.out");

inline void varf(int x)
{
	if (v[x]==x)
		return;
	varf(v[x]);
	v[x]=v[v[x]];
}

int main()
{
	int x,y,n,m;
	in>>n>>m;
	for (int i=1;i<=n;i++)
	{
		size[i]=1;
		v[i]=i;
	}
	while (m--)
	{
		in>>x;
		if (x==1)
		{
			in>>x>>y;
			varf(x);varf(y);
			if (v[x]==v[y])
				continue;
			if (size[v[x]]>size[v[y]])
			{
				v[v[y]]=v[x];
				size[v[x]]+=size[v[y]];
			}
			else
			{
				v[v[x]]=v[y];
				size[v[y]]+=size[v[x]];
			}
			continue;
		}
		in>>x>>y;
		varf(x);varf(y);
		out<<rasp[v[x]==v[y]];
	}
	return 0;
}