Cod sursa(job #1931970)

Utilizator danutbodbodnariuc danut danutbod Data 19 martie 2017 15:10:39
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.29 kb
#include <fstream>//cu reuniuni de rang si compresia drumurilor
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int T[100001],RT[100001];//vectorul de tati(T[] si vector de rang tati RT[](pentru "Reuniunea de rang")
int i,n,m,x,y,op;
int rad(int x)
{
    int r, y;
    for (r=x;T[r]!=r;r=T[r]);   //merg in sus pe arbore pana gasesc radacina
                                 //( un nod care are tata pe el insusi)
    while(T[x]!=x)       //aplic compresia drumurilor (leg directde radacina toti
      {  y = T[x]; T[x] = r;  x = y;  }  //predecesorii lui x
    return r; //returnez radacina
}
void reuneste( int x,int y)
{
    if(RT[x]>RT[y]) T[y]=x;     //unesc multimea cu rang mai mic de cea cu rang mai mare
    else T[x]=y;
    if(RT[x]==RT[y]) RT[y]++;   //in caz ca rangurile erau egale atunci cresc
}                               //rangul noii multimi cu 1
int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++) {T[i]=i; //fiecare radacina indica la ea insasi
                       RT[i]=1;//initial toti arborii au inaltimea 1
                      }
    for(i=1;i<=m;i++)
    {
        f>>op>>x>>y;
        if (op==2) { if(rad(x)==rad(y)) g<<"DA\n";
                    else g<<"NU\n";}
        else reuneste(rad(x),rad(y));
    }
    return 0;
}