Pagini recente » Cod sursa (job #337169) | Cod sursa (job #323965) | Cod sursa (job #444963) | Cod sursa (job #2374785) | Cod sursa (job #1721197)
#include <iostream>
#include <vector>
#include <fstream>
#include <cstdio>
using namespace std;
ifstream fin ("disjoint.in");
ofstream fout ("disjoint.out");
int n,m;
int RG[100020];
int TT[100020];
int find(int x)
{
int R, y;
//merg in sus pe arbore pana gasesc un nod care pointeaza catre el insusi
for (R = x; TT[R] != R; R = TT[R]);
//aplic compresia drumurilor
for (; TT[x] != x;)
{
y = TT[x];
TT[x] = R;
x = y;
}
return R;
}
void unify(int x, int y)
{
//unesc multimea cu rang mai mic de cea cu rang mai mare
if (RG[x] > RG[y])
TT[y] = x;
else TT[x] = y;
//in caz ca rangurile erau egale atunci cresc rangul noii multimi cu 1
if (RG[x] == RG[y]) RG[y]++;
}
int main()
{
fin>>n>>m;
int x,y,cd;
for (int i = 1; i <= n; i++)
{
TT[i] = i;
RG[i] = 1;
}
for (int i = 1; i <= m; i++)
{
fin>>cd>>x>>y;
if (cd == 2){
//verific daca radacina arborilor in care se afla x respectiv y este egala
if (find(x) == find(y)) fout<<"DA\n";
else fout<<"NU\n";
}
else //unesc radacinile arborilor corespunzatori multimilor nodurilor x respectiv y
{
unify(find(x), find(y));
}
}
return 0;
}