Cod sursa(job #2460938)

Utilizator stefan1anubystefan popa stefan1anuby Data 24 septembrie 2019 18:27:47
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.99 kb
#include <fstream>

using namespace std;
ifstream cin("disjoint.in");
ofstream cout("disjoint.out");

#define nmax 100005

int n,rang[nmax],mul[nmax];

int Find(int x)
{
    int R,y;
    for(R=x; R!=mul[R]; R=mul[R]);

    //R este adresa multimii care contine elementul x

    for(;mul[x]!=x;)
    {
        y=mul[x];
        mul[x]=R;
        x=y;
    }
    return R;
}

void Unite(int x, int y)
{
    if(rang[x]>rang[y])
        mul[y]=x;
    else
        mul[x]=y;

    if(rang[x]==rang[y])
        rang[y]++;
}

int main()
{
    int i,t,x,y,m;
    cin>>n>>m;
    for(i=1; i<=n; i++)
    {
        rang[i]=1;
        mul[i]=i;
    }
    for(i=1; i<=m; i++)
    {
        cin>>t>>x>>y;
        if(t==2)
        {
            if(Find(x)==Find(y))
                cout<<"DA";
            else
                cout<<"NU";
            cout<<endl;
        }
        else
        {
            Unite(Find(x),Find(y));
        }
    }
    return 0;
}