Cod sursa(job #2514616)

Utilizator anamaria_panait.10Panait Ana-Maria anamaria_panait.10 Data 26 decembrie 2019 15:26:25
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.08 kb
#include <fstream>

using namespace std;

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

int tata[100001], niv[1000001];

int cautare(int nod){
    int cpnod = nod, rad = nod;
    while(tata[rad] != 0){
        rad = tata[rad];
    }
    while(tata[cpnod] != 0){
        int aux = tata[cpnod];
        tata[cpnod] = rad;
        cpnod = aux;
    }
    return rad;
}

void reuniune(int x, int y){
    int t1 = cautare(x);
    int t2 = cautare(y);
    if(niv[t1] < niv[t2]){
        tata[t1] = t2;
    }
    else if(niv[t2] < niv[t1]){
        tata[t2] = t1;
    }
    else{
        tata[t1] = t2;
        niv[t2]++;
    }
}

bool apartinMultimii(int x, int y){
   int t1 = cautare(x);
   int t2 = cautare(y);
   return t1 == t2;
}

int main()
{
    int n, m;
    in >> n >> m;
    int operatie, x, y;
    for(int i = 1; i <= m; i++){
        in >> operatie >> x >> y;
        if(operatie == 1){
            reuniune(x, y);
        }
        else{
            apartinMultimii(x, y)?out <<"DA\n":out <<"NU\n";
        }
    }
    return 0;
}