Pagini recente » Cod sursa (job #70483) | Cod sursa (job #618347) | Cod sursa (job #2263328) | Cod sursa (job #2760670) | Cod sursa (job #2944485)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream in("disjoint.in");
ofstream out("disjoint.out");
int n, m, op, x, y;
vector<int> repr, rg;
int findRepr(int x)
{
while (x != repr[x])
x = repr[x];
return x;
// urc pana gasesc reprezentantul lui x
}
void join(int x, int y)
{
int rx = findRepr(x), ry = findRepr(y);
// extrag reprezentatii lui x si lui y
if (rg[rx] < rg[ry])
repr[rx] = ry;
else if (rg[rx] > rg[ry])
repr[ry] = rx;
// reprezentantul arborelui cu rang mai mic = reprezentantul arborelui cu
// rang mai mare
else
{
repr[rx] = ry;
rg[ry]++;
// daca ambele au rang egal nu conteaza ordinea, dar trebuie sa cresc
// rangul celui care devine reprezentant
}
}
int main()
{
in >> n >> m;
repr.resize(n + 1, 0);
rg.resize(n + 1, 0);
for (int i = 1; i <= n; i++)
repr[i] = i;
while (m--)
{
in >> op >> x >> y;
if (op == 1)
join(x, y);
else if (op == 2)
out << ((findRepr(x) == findRepr(y)) ? "DA\n" : "NU\n");
}
return 0;
}