Cod sursa(job #2932353)

Utilizator StanCatalinStanCatalin StanCatalin Data 2 noiembrie 2022 18:39:33
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.75 kb
#include <iostream>
#include <fstream>

using namespace std;

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

const int dim = 1e5 + 5;

int parent[dim], rang[dim], n, m;

int Find(int u) {
    while (parent[u] != 0) {
        u = parent[u];
    }
    return u;
}

void Union(int u, int v) {
    int ru, rv;
    ru = Find(u);
    rv = Find(v);

    if (rang[ru] < rang[rv]) {
        parent[ru] = rv;
    } else {
        parent[rv] = ru;
        if (rang[ru] == rang[rv]) {
            rang[rv]++;
        }
    }
}

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

    int tip, x, y;
    while (m--) {
        in >> tip >> x >> y;

        if (tip == 1) Union(x, y);
        else {
            out << ((Find(x) == Find(y)) ? "DA\n" : "NU\n");
        }
    }
    return 0;
}