Pagini recente » Utilizatori inregistrati la PreOJI 2017 | Cod sursa (job #757716) | Cod sursa (job #1843837) | Cod sursa (job #3238163) | Cod sursa (job #2198305)
#include <iostream>
#include <fstream>
#include <vector>
#include <cmath>
#include <algorithm>
#define NMAX 100020
using namespace std;
int TT[NMAX], RG[NMAX];
int N, M;
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 unite(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()
{
ifstream in("disjoint.in");
ofstream out("disjoint.out");
in >>N>>M;
int x,y,cd,i;
//initial fiecare nod pointeaza catre el insusi iar rangul fiecarui arbore este 1
for (i = 1; i <= N; i++)
{
TT[i] = i;
RG[i] = 1;
}
for (i = 1; i <= M; i++)
{
in >>cd>>x>>y;
if (cd == 2){
///verific daca radacina arborilor in care se afla x respectiv y este egala
if (Find(x) == Find(y)) out<<"DA"<<"\n";
else out <<"NU"<<"\n";
}
else ///unesc radacinile arborilor corespunzatori multimilor nodurilor x respectiv y
{
if (Find(x) == Find(y)) {out <<i;return 0;}
unite(Find(x), Find(y));
}
}
return 0;
}