Cod sursa(job #2471685)

Utilizator IoanStoicaStoica Ioan IoanStoica Data 11 octombrie 2019 10:50:50
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.11 kb
///Var 3
///Reuniunea dupa rang: Pentru fiecare multime tinem minte inaltimea arborelui care reprezinta acea multime
///si atunci cand vrem sa unim 2 arbori, il unim pe cel mai mic de cel mai mare.
///Compresia drumurilor
#include <bits/stdc++.h>
#define K 100002
using namespace std;

ifstream f("disjoint.in");
ofstream g("disjoint.out");

int t[K],h[K];
int tata (int x)
{
    if(t[x]==0)
        return x;
    return tata(t[x]);
}
void compresie(int tt,int fiu)
{
    if(fiu==tt)return ;
    t[fiu]=tt;
    compresie(tt,t[fiu]);
}
int main()
{
    int i,n,m,cod,x,y,tx,ty;
    f>>n>>m;
    for(i=1;i<=n;i++)
        t[i]=0,h[i]=1;
    while(m--)
    {
        f>>cod>>x>>y;
        if(cod==2)
            if(tata(x)==tata(y))
                g<<"DA\n";
            else g<<"NU\n";
        else
        {
            tx=tata(x);ty=tata(y);
            if(h[tx] < h[ty])///Reuniunea dupa rang:
            {
                swap(tx,ty);
                swap(x,y);
            }
            t[ty]=tx;
            h[tx]=max(h[tx],h[ty]+1);
            compresie(tx,y);
        }
    }
}