Cod sursa(job #2170025)

Utilizator TudorFinaruTudor Cristian Finaru TudorFinaru Data 14 martie 2018 21:02:30
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.91 kb
#include <fstream>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int r[100003],p[100003],n,m;

int Find(int x)
{
    if(x!=p[x]) p[x]=Find(p[x]);
    return p[x];
}

void Union(int x, int y)
{
    int xroot,yroot;
    xroot=Find(x);
    yroot=Find(y);
    if(r[xroot]<=r[yroot])
    {
        r[yroot]+=r[xroot];
        p[xroot]=p[yroot];
    }
    else
    {
        r[xroot]+=r[yroot];
        p[yroot]=p[xroot];
    }
}


int main()
{
    int i,cod,x,y;
    ///p[] - parent ; r[] - rank
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        p[i]=i;
        r[i]=1;
    }
    for(i=1;i<=m;i++)
    {
        f>>cod>>x>>y;
        if(cod==1)
        {
            Union(x,y);
        }
        else
        {
            if(Find(x)!=Find(y)) g<<"NU"<<'\n';
            else g<<"DA"<<'\n';
        }
    }
    f.close();
    g.close();
    return 0;
}