Cod sursa(job #2941300)

Utilizator Cosmin3105Cosmin Colceru Cosmin3105 Data 17 noiembrie 2022 16:54:27
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.92 kb
#include <iostream>
#include <fstream>

using namespace std;

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

const int NMAX = 100005;
int n, m;
int tata[NMAX], rang[NMAX];

int radacina(int x) {
	while (tata[x] != x) {
		x = tata[x]; 
	}

	return x;
}

void reuniune(int x, int y) {
	int radacinaX = radacina(x);
	int radacinaY = radacina(y);

	if (rang[radacinaX] < rang[radacinaY])
		tata[radacinaX] = radacinaY;
	else if (rang[radacinaX] > rang[radacinaY])
		tata[radacinaY] = radacinaX;
	else {
		tata[radacinaX] = radacinaY;
		rang[radacinaY]++;
	}
}

int main() {
	fin >> n >> m;
	for (int i = 1; i <= n; i++) {
		tata[i] = i;
		rang[i] = 1;
	}

	for (int i = 0; i < m; i++) {
		int cod, x, y;
		fin >> cod >> x >> y;
		if (cod == 1) {
			reuniune(x, y);
		}
		else {
			if (radacina(x) == radacina(y))
				fout << "DA" << endl;
			else 
				fout << "NU" << endl;
		}
	}
}