Cod sursa(job #2864796)

Utilizator OrosIacobOros Iacob OrosIacob Data 8 martie 2022 11:21:46
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.23 kb
#include <iostream>
#include <fstream>

#define NMAX 100020

using namespace std;

 ifstream f("disjoint.in");
 ofstream g("disjoint.out");

int parinte[NMAX], marime[NMAX];
int n, m;

int sef(int x)
{
    int radacina=x, y;

    while(radacina!=parinte[radacina])
        radacina=parinte[radacina];

    //compresie
    while(x!=radacina)
    {

        y = parinte[x];
        parinte[x] = radacina;
        x = y;
    }
    return radacina;
}

void unite(int x, int y)
{
    x=sef(x);
    y=sef(y);
    parinte[x]=y;

}

int main()
{
    f>>n>>m;
    int i, x, y, p;

    //initial fiecare nod pointeaza catre el insusi iar rangul fiecarui arbore este 1
    for (i=1; i<=n; i++)
    {
        parinte[i]=i;
        marime[i]=1;
    }

    for (i=1; i<=m; i++)
    {
        f>>p>>x>>y;

        if (p==2)
        {
            //verific daca radacina arborilor in care se afla x respectiv y este egala
            if (sef(x) == sef(y))
                g<<"DA\n";
            else
                g<<"NU\n";
        }
        if(p==1) //unesc radacinile arborilor corespunzatori multimilor nodurilor x respectiv y
        {

            unite(x, y);
        }
    }

    return 0;
}