Cod sursa(job #2572810)

Utilizator smoc_georgemarianSmoc George-Marian smoc_georgemarian Data 5 martie 2020 14:24:46
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <bits/stdc++.h>
#define NMAX 100009
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int tata[NMAX];
int h[NMAX];
int n,m,i;
void citire();
int aflare(int x);
void unire(int x, int y);
int main()
{citire();
    return 0;
}
void citire()
{ int x,y,t,cx,cy;
  fin>>n>>m;
  for(i=1;i<=m;i++)
    {
     fin>>t>>x>>y;
     cx=aflare(x);
     cy=aflare(y);
     if(t==1)
        {
         unire(cx,cy);
        }
        else
        {
          if(cx==cy)
                fout<<"DA"<<'\n';
               else
                fout<<"NU"<<'\n';
        }
    }
}
int aflare(int x)
{
  int rad,cop;
  cop=x;
  while(tata[x])
        x=tata[x];
  rad=x;///tatal tatilor
  ///facem drumul scurt
  while(tata[cop])
        {
         x=cop;
         cop=tata[cop];
         tata[x]=rad;
        }
   return rad;
}
void unire(int x, int y)
{
  if(h[x]>h[y])
        tata[y]=x;
  else
  {
    if(h[y]>h[x])
        tata[x]=y;
    else
    {tata[x]=y;h[y]++;}
  }
}