Cod sursa(job #3330107)

Utilizator marelucaMare Luca Ghita mareluca Data 17 decembrie 2025 16:42:05
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.12 kb
#include <bits/stdc++.h>

std::ifstream fin("disjoint.in");
std::ofstream fout("disjoint.out");

int n, m;
std::vector<int> radacina, rang;

int find_R(int node) {
    int R;

    // Cautam radacina
    for(R = node;
        radacina[R] != R;
        R = radacina[R])

    // Compresia drumului
    while(radacina[node] != node) {
        int r = radacina[node];
        radacina[node] = R;
        node = r;
    }

    return R;
}

void unite(int node_A, int node_B) {
    if(rang[node_A] > rang[node_B])
        radacina[node_B] = radacina[node_A];
    else
        radacina[node_A] = radacina[node_B];

    if(rang[node_A] == rang[node_B]) ++rang[node_B];
}

int main() {
    fin >> n >> m;

    radacina.resize(n + 1);
    rang.resize(n + 1);

    for(int i = 1; i <= n; ++i) {
        radacina[i] = i;
        rang[i] = 1;
    }

    while(m--) {
        int op, x, y;
        fin >> op >> x >> y;

        if(op == 1) {
            unite(find_R(x), find_R(y));
        }
        else {
            if(find_R(x) == find_R(y))
                fout << "DA\n";
            else
                fout << "NU\n";
        }
    }

    return 0;
}