Cod sursa(job #2525957)

Utilizator smoc_georgemarianSmoc George-Marian smoc_georgemarian Data 18 ianuarie 2020 09:35:18
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <bits/stdc++.h>
#define NMAX 100009
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int tata[NMAX],  h[NMAX]; ///inalt arborelui cu rad i
int n;
//fct
int Find(int x); ///return la radacina arborelui caruia apartine x
void Union(int x, int y); ///reuneste cu arborele cu rad x si cel cu raad y
int m,i;
int main()
{fin>>n>>m;
 for(i=1;i<=m;i++)
    {
     int x,y,c;
     int cx,cy;
     fin>>c>>x>>y;
     cx=Find(x);
     cy=Find(y);
     if(c==2)
        {if(cx!=cy)
             fout<<"NU"<<'\n';
             else
                fout<<"DA"<<'\n';
        }
       else
            if(cx!=cy)
              Union(cx,cy);
    }


    return 0;
}
int Find(int x)
{
int rad,aux;
rad=x;
while(tata[rad]) rad=tata[rad];
///fac compresia drumului
while(tata[x])
    {aux=tata[x];
     tata[x]=rad;
     x=aux;
    }
return rad;
///O(n)
}
void Union(int x, int y)
{
 tata[y]=x; ///O(1)
 if(h[x]<h[y])
    tata[x]=y;
    else
    {tata[y]=x;
     if(h[x]==h[y])
        h[x]++;

    }

}