Cod sursa(job #2944485)

Utilizator popescumateicalinPopescu Matei Calin popescumateicalin Data 22 noiembrie 2022 16:55:54
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.19 kb
#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;
}