Cod sursa(job #2942649)

Utilizator adeladanescuAdela Danescu adeladanescu Data 19 noiembrie 2022 22:27:12
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.46 kb
#include <iostream>
#include <fstream>
using namespace std;
#define NMAX 100001
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int n,m,op,x,y,rad_X, rad_Y, rang[NMAX],radacina[NMAX];

int cautaRadacina(int x)
{
    int R=x;
    while(radacina[R] != R) //aflam in ce multime se afla nodul x
    {
        R=radacina[R];
    }
    while(radacina[x]!=x)  //mai parcurgem o data drumul de la x la radacina si unim toate nodurile direct de radacina
    {
        int aux=radacina[x];
        radacina[x]=R;
        x=aux;
    }
    return R; //intoarcem radacina
}

void reuniune(int x, int y)
{
    if(rang[x]==rang[y])
        rang[y]++; //tinem minte inaltimea multimii
    if(rang[x]<rang[y])
        swap(x,y);
    radacina[y]=x; //unim arborele mai mic de arborele mai mare

}

void operatii(int op, int x, int y)
{
    rad_X= cautaRadacina(x);
    rad_Y= cautaRadacina(y);
    if(op==1)
    {
        if(rad_X!=rad_Y)  //daca cele doua elemente nu se afla in aceeasi multime, le reunim
            reuniune(rad_X, rad_Y);
    }
    if(op==2){
        if(rad_X==rad_Y) //daca se afla in aceeasi multime Da, altfel Nu
            fout << "DA" << endl;
        else
            fout << "NU" << endl;
    }
}
int main()
{
    fin >> n >> m;
    for(int i=1;i<=n;i++)
    {
        //construim multimile si le retinem inaltimile
        radacina[i]=i;
        rang[i]=1;
    }
    for(int i=1;i<=m;i++)
    {
        fin >> op >> x >> y;
        operatii(op,x,y);
    }
    return 0;
}