Cod sursa(job #2948667)

Utilizator readyitzooDilirici Mihai readyitzoo Data 27 noiembrie 2022 22:58:59
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.91 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

int n, m;
int x, y, op;

vector < int > tata; // vector de tati

int getRoot(int &a) {   // functie recursiva pentru a afla radacina unui nod
    if (tata[a] == a) {
        return a;
    }

    return getRoot(tata[a]);
}

int main() {
    f >> n >> m;        // citire date
    tata.resize(n + 1);

    for (int i = 1; i <= n; i++)
        tata[i] = i;        // deoarece toate nodurile formeaza singure o comp conexa, initializam vectorul de tati cu tata[i] = i

    for (int i = 0; i < m; i++)
    {
        f >> op >> x >> y;      // citire date
        if (op == 1) {
            tata[getRoot(y)] = getRoot(x);
        } else {
            if (getRoot(x) == getRoot(y)) {
                g << "DA" << '\n';
            } else {
                g << "NU" << '\n';
            }
        }
    }
}