Cod sursa(job #2432504)

Utilizator monaaMona Dumbravescu monaa Data 23 iunie 2019 23:20:58
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.82 kb
#include <fstream>
#define NMAX 100100
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int grad[NMAX], tata[NMAX];
int n, m, i, a, b, c;
int find(int nod)
{
	if(tata[nod] == nod) return nod;
	tata[nod] = find(tata[nod]);
	return tata[nod];
}
void unionn(int x, int y)
{
	int tx, ty;
	tx = find(x);
	ty = find(y);
	if(tx != ty)
	{
		if (grad[tx] >= grad[ty])
		{
			tata[ty] = tx;
			grad[tx] = grad[tx] + grad[ty];
		}
		else
		{
			tata[tx] = ty;
			grad[ty] = grad[ty] + grad[tx];
		}
	}
}
int main()
{
	f >> n >> m;
	for(i = 1; i <= n; i ++)
	{
		tata[i] = i;
		grad[i] = 0;
	}
	for(i = 1; i <= m; i ++)
	{
		f >> a >> b >> c;
		if(a == 1)
			unionn(b, c);
		else
			if(find(b) == find(c))
				g << "DA" << "\n";
			else
				g << "NU" << "\n";
	}
	return 0;
}