Cod sursa(job #2941314)

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

using namespace std;

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

int n, m;
vector<int> tata;

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

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

	return r;
}

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

	for (int i = 0; i < m; i++) {
		int cod, x, y;
		fin >> cod >> x >> y;
		x = radacina(x);
		y = radacina(y);

		if (cod == 1) {
			tata[y] = x;		
		}
		else {
			if (x == y)
				fout << "DA" << endl;
			else 
				fout << "NU" << endl;
		}
	}
}