Cod sursa(job #2750914)

Utilizator AlinSSimilea Alin-Andrei AlinS Data 13 mai 2021 16:44:59
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.34 kb
#include <fstream>

using namespace std;

ifstream fin("disjoint.in");
ofstream fout("disjoint.out");

const int NMAX=10000;
int t[NMAX+5],h[NMAX+5];

int FindSet(int x)
{
    while(t[x]!=x)
        x=t[x];
    return x;
}

void UnionSet(int x, int y)
{
    /*x,y - radacinile celor doi arbori care se unifica*/

    if(h[x]>h[y])
    {
        t[y]=x;
    }
    else if(h[y]>h[x])
    {
        t[x]=y;
    }
    else
    {
        t[y]=x;
        h[x]++;
    }

}

int main()
{
    int n,m,i,j,x,y,cod,rx,ry,nr=0;
    fin>>n>>m;

    for(i=1; i<=n; i++)
    {
        t[i]=i;
        h[i]=1;
    }
    for(i=1; i<=m; i++)
    {
        fin>>cod>>x>>y;
        if(cod==1)
        {
            rx=FindSet(x);
            ry=FindSet(y);
            if(rx!=ry) UnionSet(rx,ry);
        }
        if(cod==2)
        {
            rx=FindSet(x);
            ry=FindSet(y);
            if(rx==ry) fout<<"DA"<<endl;
            else fout<<"NU"<<endl;
        }

    }

    /*
        for(i=1; i<=n; i++)
            if(t[i]==i)
            {
                fout<<"Arbore cu radacina "<<i<<":"<<i<<" ";
                for(j=1; j<=n; j++)
                    if(j!=i)
                        if(FindSet(j)==i) fout<<j<<" ";
                fout<<"\n";
            }
    */

    return 0;
}