Cod sursa(job #3208154)

Utilizator alexlazuLazureanu Alexandru Ioan alexlazu Data 27 februarie 2024 21:40:59
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.45 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <unordered_map>
#include <vector>
#include <tuple>
#include <queue>
#define inmat lin >= 1 && lin <= n && col >=1 && col <= m

using namespace std;

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

#define cin in
#define cout out


const int N = 1e5+2000;

int n, q;

int rad[N] , rang[N];

void unite(int x, int y)
{
	//unesc multimea cu rang mai mic de cea cu rang mai mare
	if (rang[x] > rang[y])
		rad[y] = x;
	else rad[x] = y;

	//in caz ca rangurile erau egale atunci cresc rangul noii multimi cu 1
	if (rang[x] == rang[y]) rang[y]++;
}

int find(int x)
{
	int R, y;

	//merang in sus pe arbore pana gasesc un nod care pointeaza catre el insusi
	for (R = x; rad[R] != R; R = rad[R]);

	//aplic compresia drumurilor
	for (; rad[x] != x;)
	{
		y = rad[x];
		rad[x] = R;
		x = y;
	}
	return R;
}

int main()
{
	cin >> n >> q;
	for (int i = 1; i <= n; i++)
	{
		rad[i] = i;
		rang[i] = 1;
	}
	int cd, x, y;
	for (int i = 1; i <= q; i++)
	{
		cin >> cd >> x >> y;

		if (cd == 2) {
			//verific daca radacina arborilor in care se afla x respectiv y este egala
			if (find(x) == find(y)) cout << "DA\n";
			else cout << "NU\n";
		}
		else //unesc radacinile arborilor corespunzatori multimilor nodurilor x respectiv y
		{
			if (find(x) == find(y)) { fprintf(stderr, "%d ", i); return 0; }
			unite(find(x), find(y));
		}
	}
}