Cod sursa(job #2907469)

Utilizator maria.ianiIani Maria maria.iani Data 30 mai 2022 13:24:21
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.5 kb
#include <iostream>
#include <fstream>

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

#define MAX 100001

int N,M,cod,x,y,i;
int  tata[MAX], rang[MAX];

int search(int x)
{
    int radacina,aux;
    //caut un nod care pointeaza catre el insusi
    radacina = x;
    while (radacina != tata[radacina])
        radacina = tata[radacina];

    // compresia drumurilor
    for (; tata[x] != x;) {
        aux = tata[x];
        tata[x] = radacina;
        x = aux;
    }
    return radacina;

}

void unite(int x, int y)
{
    //cand vrem sa unim 2 arbori, il unim pe cel mai mic de cel mai mare
    if (rang[x] > rang[y])
        tata[y] = x;
    else
        tata[x] = y;

    //daca rangurile sunt egale cresc rangul noii multimi cu 1
    if (rang[x] == rang[y])
        rang[y]++;
}

int main()
{

    f >> N >> M;

    // initializez vectorul de tati cu nodul insusi
    // rangul fiecarui nod estee 1
    for (i = 1; i <= N; i++)
    {
        tata[i] = i;
        rang[i] = 1;
    }

    for (i = 1; i <= M; i++)
    {
        f >> cod >> x >> y;
        if (cod == 2){
            // verific daca nodurile citite au aceeasi radacina
            if (search(x) == search(y))
                g<<"DA"<<endl;
            else
                g<<"NU"<<endl;
        }
        else //unesc radacinile arborilor corespunzatori multimilor nodurilor x respectiv y
        {
            if (search(x) == search(y))
                return 0;

            unite(search(x), search(y));
        }
    }

    return 0;
}