Cod sursa(job #3182671)

Utilizator FilipGeorgeBumbar Filip George FilipGeorge Data 9 decembrie 2023 12:38:02
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.82 kb
// https://infoarena.ro/problema/disjoint

#include <fstream>

using namespace std;

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

int i, p[100001], Rank[100001], cod, x, y, n, m, o; // rank este altceva

void Union (int r1, int r2) { // Se reunesc multimile cu radacinile r1 si r2.
  if (Rank[r1] > Rank[r2])
    p[r2] = r1;
  else {
    p[r1] = r2;
    if (Rank[r1] == Rank[r2])
      Rank[r2]++;
  }
}

int FindSet (int nod) {
  while (p[nod] != nod)
    nod = p[nod];
  return p[nod];
}

int main () {
  fin >> m >> o;
  for (i = 1; i <= m; i++)
    p[i] = i;
  while (o--) {
    fin >> cod >> x >> y;
    if (cod == 1)
      Union(FindSet(x), FindSet(y));
    else
      if (FindSet(x) == FindSet(y))
        fout << "DA\n";
      else
        fout << "NU\n";
  }
  return 0;
}