Cod sursa(job #2403172)

Utilizator Marius92roMarius Marius92ro Data 11 aprilie 2019 12:30:46
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.72 kb
#include <iostream>
#include <fstream>

using namespace std;

int vectorTati[100005];
int rang[100005];

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

int radacina(int nod){

    int k = nod;

    while ( vectorTati[k] != 0 ) //caut radacina nodului
            k = vectorTati[k];

    while ( vectorTati[nod] != 0 ){ //actualizez vectorul de tati cu radacina gasita
         vectorTati[nod] = k;
         nod = vectorTati[nod];
    }

    return nod;

}

void unire(int x, int y){ //leg radacinile a  doua componente conexe

    int radacinaX = radacina(x);
    int radacinaY = radacina(y);

    if ( radacinaX != radacinaY ){ //in caz ca radacinile sunt diferite -> avem doua submultimi ( componente conexe ) diferite

            if(rang[radacinaX] > rang[radacinaY])  //unesc multimea cu rang mai mic de cea cu rang mai mare
                        vectorTati[radacinaY] = radacinaX;

            else
                vectorTati[radacinaX] = radacinaY;


            if( rang[radacinaX] == rang[radacinaY] ) //in caz ca rangurile erau egale atunci cresc rangul noii multimi cu 1
                                    rang[radacinaY] ++;
    }

}


int main(){  // problema de pe info arena   https://infoarena.ro/problema/disjoint

    int nrNoduri , nrOperatii , x , y , cod;

    fin >> nrNoduri >> nrOperatii;

    int nrComponenteConexe = nrNoduri;

    for (int i = 1; i <= nrOperatii; i++){

        fin >> cod >> x >> y;

        if ( cod == 1 ){
            unire(x,y);
            nrComponenteConexe--;
        }

        else {
            if ( radacina(x) == radacina(y) ) fout<<"DA"<<endl;
            else fout<<"NU"<<endl;

        }
    }

    return 0;

}