Cod sursa(job #1701494)

Utilizator misu012Pop Mihai misu012 Data 13 mai 2016 11:36:18
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include <fstream>
using namespace std;

ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m,x,y,c,t[100055],r[100055];

int gasire(int x)
{
    for(x=t[x];x!=t[x];x=t[x])
        return x;
}

int gasire_radacina(int x)
{
    if(t[x]==x)
        return x;
    else
        return t[x] = gasire_radacina(t[x]);
}


void reuniune(int x,int y)
{
    int a,b;
    a=gasire_radacina(x);
    b=gasire_radacina(y);

    if(r[a]<r[b])
        t[a]=b;
    else
        t[b]=a;

    if(r[a]==r[b])
        ++r[a];

}

int main()
{

    f>>n>>m;
    for(int i=1;i<=n;i++)
    {
        t[i]=i;
        r[i]=0;
    }

    for(int i=1;i<=m;i++)
    {
        f>>c>>x>>y;
        if(c==1)
            reuniune(x,y);
        else
            if(gasire_radacina(x)==gasire_radacina(y))
                g<<"DA\n";
            else
                g<<"NU\n";
        }


    return 0;
}