Cod sursa(job #2701407)

Utilizator Gota_AndreiGota Andrei Gota_Andrei Data 30 ianuarie 2021 22:10:20
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.79 kb
#include<bits/stdc++.h>
using namespace std;

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

const int NMAX = 100005;
int father[NMAX];
int rang[NMAX];
int n, m, i, x, y, condition;

int find_root(int x) {
  if (father[x] == 0)
    return x;
  else {
    int r = find_root(father[x]);
    father[x] = r;
    return r;
  }
}

int link_nodes(int x, int y) {
  if (rang[x] > rang[x])
    father[y] = x;
  else
    father[x] = y;

  if (rang[x] == rang[y])
    rang[y]++;
}

int main() {
  fin >> n >> m;
  while (m--) {
    fin >> condition >> x >> y;
    if (condition == 1)
      link_nodes(find_root(x), find_root(y));
    else {
      if (find_root(x) == find_root(y))
        fout << "DA" << endl;
      else
        fout << "NU" << endl;
    }
  }
}