Pagini recente » Cod sursa (job #990203) | Cod sursa (job #614187) | Cod sursa (job #480603) | Cod sursa (job #246202) | Cod sursa (job #2412238)
#include <iostream>
#include <fstream>
#include <stdlib.h>
using namespace std;
int *vectorTati, *rang;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int radacina(int nod){
int radacinaNod = nod;
while ( vectorTati[radacinaNod] != radacinaNod ) //caut radacina nodului
radacinaNod = vectorTati[radacinaNod];
while ( vectorTati[nod] != nod ){ //actualizez vectorul de tati cu radacina gasita
int aux = vectorTati[nod];
vectorTati[nod] = radacinaNod;
nod = aux;
}
return radacinaNod;
}
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(){
int nrNoduri , nrOperatii , x , y , cod;
fin >> nrNoduri >> nrOperatii;
vectorTati = NULL;
rang = NULL;
vectorTati = new int[nrNoduri + 1];
rang = new int[nrNoduri + 1];
if ( !vectorTati || !rang ){
cout << "\nEroare la alocare dinamica !\n";
exit(-1);
}
for ( int i = 1; i <= nrNoduri; i++){ // intitial fiecare nod este un nod izolat
vectorTati[i] = i;
rang[i] = 1;
}
for (int i = 1; i <= nrOperatii; i++){
fin >> cod >> x >> y;
if ( cod == 1 )
unire(radacina(x),radacina(y));
else {
if ( radacina(x) == radacina(y) )
fout<<"DA\n";
else
fout<<"NU\n";
}
}
//for ( int i = 1; i <= nrNoduri; i++)
// cout << vectorTati[i] << " ";
return 0;
}