Cod sursa(job #2948187)

Utilizator AndreiBerbecaruBerbecaru-Iovan Andrei AndreiBerbecaru Data 27 noiembrie 2022 13:40:19
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.24 kb
///Idee: retinem multimile in vectorul tata, avand initial rangul 1.
///Dupa aceea citim operatiile si apelam corespunaztor codului functiile
///Daca avem functia reuniune atunci pt fiecare multime, folosind functia CautaRadacina, retinem tatal elementului.
///Cea care are gradul mai mare este tata pentru cealalta
///Daca avem functia afiseaza verificam daca multimile au acelasi tata
///Complexitate:O(n^2 + mlogn)

#include<fstream>
#include<vector>
using namespace std;

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

int n, m, cod, x, y;
vector<int> tata;

int cautaRadacina(int n){
    while(tata[n] != n){
        tata[n] = tata[tata[n]];
        n = tata[n];
    }

    return n;
}

void reuniune(int& x, int& y){
    int a = cautaRadacina(x);
    int b = cautaRadacina(y);

    tata[a] = tata[b];
}

void afisare(int& x, int& y){
    if (cautaRadacina(x) == cautaRadacina(y))
        fout<<"DA"<<"\n";
    else
        fout<<"NU"<<"\n";
}

int main(){

    fin>>n>>m;
    tata.resize(n+1);

    for(int i=1; i<=n; i++)
        tata[i] = i;

    for(int i=1; i<=m; i++){
        fin>>cod>>x>>y;
        if(cod == 1)
            reuniune(x,y);
        else
            afisare(x,y);
    }

}