Cod sursa(job #2342201)

Utilizator raxman01Sicoe Raul Ioan raxman01 Data 12 februarie 2019 17:39:39
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.96 kb
#include <fstream>

using namespace std;

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

int tata[100002],n,m,x,y,p,rg[100002];

inline int findt (int nod)
{
    int aux;
    while (nod!=tata[nod])
    {
       aux=nod;
       nod=tata[nod];
       tata[aux]=tata[nod];
    }
    return nod;
}

inline void unire (int a,int b)
{
    if (rg[a]>rg[b])
        tata[b]=a;
    if (rg[b]>rg[a])
        tata[a]=b;
    if (rg[b]==rg[a])
    {
        tata[a]=b;
        rg[a]++;
    }
}

int main()
{
    int p1,p2;
    fin>>n>>m;
    for (int i=1;i<=n;i++)
        tata[i]=i;
   for (int i=1;i<=m;i++)
   {
       fin>>p>>x>>y;
        p1=findt(x);
        p2=findt(y);
        if (p==1)
        {
            if (p1!=p2)
                unire(p1,p2);
        }
        else
        {
            if (p1==p2)
                fout<<"DA\n";
            else
                fout<<"NU\n";
        }
   }
    return 0;
}