Pagini recente » Cod sursa (job #527791) | Cod sursa (job #1928752) | Cod sursa (job #1879476) | Cod sursa (job #1648160) | Cod sursa (job #2936937)
/// Ne vom gandi la fiecare multime ca la un arbore
/// Initial avem n arbori a cate un nod fiecare
/// La fiecare unificare, vom lega arborele mai mic la cel mai mare
#include <fstream>
#include <vector>
int main()
{
std::ifstream fin("disjoint.in");
std::ofstream fout("disjoint.out");
int n, m;
fin >> n >> m;
/// tatal fiecarui nod
/// ancestor[x] = x inseamna ca x este radacina arborelui respectiv
/// initial avem n arbori, asadar ancestor[i] = i, i->{1..n}
std::vector<int> ancestor(n + 1);
for (int i = 1; i <= n; ++i) ancestor[i] = i;
/// pentru a lega arborele mai mic de arborele mai mare
/// vom retine rangul fiecarui nod (initial 1)
std::vector<int> rank(n + 1, 1);
for (int i = 0; i < m; ++i) {
int op, x, y;
fin >> op >> x >> y;
/// gasim radacinile arborilor din care face parte x si y
while (ancestor[x] != x || ancestor[y] != y) {
if (ancestor[x] != x) x = ancestor[x];
if (ancestor[y] != y) y = ancestor[y];
}
/// legam arborele mai mic la cel mai mare
/// daca sunt egale, legam oricum
if (op == 1) {
if (rank[x] < rank[y]) ancestor[x] = y;
else if (rank[x] > rank[y]) ancestor[y] = x;
else ancestor[x] = y, ++rank[y];
}
/// daca au aceeasi radacina, fac parte din acelasi arbore
else fout << (x == y ? "DA" : "NU") << '\n';
}
return 0;
}