Pagini recente » Cod sursa (job #497408) | Cod sursa (job #646140) | Cod sursa (job #371960) | Cod sursa (job #826674) | Cod sursa (job #2907471)
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
#define MAX 100001
int N,M,cod,x,y,i;
int tata[MAX], rang[MAX];
int search(int x)
{
int radacina,aux;
//caut un nod care pointeaza catre el insusi
for (radacina = x; tata[radacina] != radacina; radacina = tata[radacina]);
// compresia drumurilor
for (; tata[x] != x;) {
aux = tata[x];
tata[x] = radacina;
x = aux;
}
return radacina;
}
void unite(int x, int y)
{
//cand vrem sa unim 2 arbori, il unim pe cel mai mic de cel mai mare
if (rang[x] > rang[y])
tata[y] = x;
else
tata[x] = y;
//daca rangurile sunt egale cresc rangul noii multimi cu 1
if (rang[x] == rang[y])
rang[y]++;
}
int main()
{
f >> N >> M;
// initializez vectorul de tati cu nodul insusi
// rangul fiecarui nod estee 1
for (i = 1; i <= N; i++)
{
tata[i] = i;
rang[i] = 1;
}
for (i = 1; i <= M; i++)
{
f >> cod >> x >> y;
if (cod == 2){
// verific daca nodurile citite au aceeasi radacina
if (search(x) == search(y))
g<<"DA"<<endl;
else
g<<"NU"<<endl;
}
else //unesc radacinile arborilor corespunzatori multimilor nodurilor x respectiv y
{
if (search(x) == search(y))
return 0;
unite(search(x), search(y));
}
}
return 0;
}