Cod sursa(job #2938114)

Utilizator woodyinhoStoica Elias-Valeriu woodyinho Data 11 noiembrie 2022 17:02:14
Problema Paduri de multimi disjuncte Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.2 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;

ifstream f("disjoint.in");
ofstream g("disjoint.out");

int n, m;
vector<int> parinti;
vector<int> ranc;

int get_root(int x) {
    if (parinti[x] == x) {
        return x;
    }
    return parinti[x] = get_root(parinti[x]);
}

int main() {
    f >> n >> m;
    parinti.resize(n + 1);
    ranc.resize(n + 1);
    for(int i = 1; i <= n; i++){
        parinti[i] = i;
        ranc[i] = 1;
    }
    for (int i = 1; i <= m; i++) {
        int cod, x, y;
        f >> cod >> x >> y;
        if (cod == 1) {
            int rx = get_root(x);
            int ry = get_root(y);
            if (rx != ry) {
                if (ranc[rx] > ranc[ry]) {
                    parinti[ry] = rx;
                    ranc[rx] += ranc[ry];
                } else {
                    parinti[rx] = ry;
                    ranc[ry] += ranc[rx];
                }
            }
        }
        else {
            int rx = get_root(x);
            int ry = get_root(y);
            if (rx == ry) {
                g << "DA" << endl;
            }
            else {
                g << "NU" << endl;
            }
        }
    }
    return 0;
}