Cod sursa(job #2493506)

Utilizator VladAdrianaVlad Adriana VladAdriana Data 16 noiembrie 2019 13:19:04
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.71 kb
#include <fstream>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int tata[100005], n, m,h[100005],op,x,y;
int Find(int k)
{
	int x = k;
	while (tata[k])
		k = tata[k];
	if (x != k)
	{
		while (tata[x] != k)
		{
			int aux = tata[x];
			tata[x] = k;
			x = aux;
		}
	}
	return k;
}
void Union(int x, int y)
{
	x = Find(x); 
	y = Find(y);
	if (h[x] < h[y])
		tata[x] = y;
	else if(h[x]>h[y])
		tata[y] = x;
	else
	{
		tata[x] = y;
		h[y]++;
	}
}
int main()
{
	int i;
	fin >> n >> m;
	for(i=1;i<=m;i++)
	{
		fin >> op>>x>>y;
		if (op == 1) Union(x, y);
		else
		{
			if (Find(x) == Find(y)) fout << "DA\n";
			else fout << "NU\n";
		}
	}
	return 0;
}