Cod sursa(job #1729386)

Utilizator ValentinSavoiuFMI Savoiu Valentin-Marian ValentinSavoiu Data 14 iulie 2016 17:12:29
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.71 kb
#include <fstream>

using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int tata[100001],rang[100001],i,n,m,x,y,t;
int rad(int x)
{
    int r=x,aux;
    while(tata[r]!=r) r=tata[r];
    while(x!=tata[x])
    {
        aux=tata[x];
        tata[x]=r;
        x=aux;
    }
    return r;
}
void reun( int x,int y)
{
    if(rang[x]>rang[y]) tata[y]=x;
    else tata[x]=y;
    if(rang[x]==rang[y]) rang[y]++;
}
int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++) {tata[i]=i; rang[i]=1;}
    for(i=1;i<=m;i++)
    {
        f>>t>>x>>y;
        if (t==2) { if(rad(x)==rad(y)) g<<"DA\n";
                    else g<<"NU\n";}
        else reun(rad(x),rad(y));
    }
    return 0;
}