Cod sursa(job #3370490)

Utilizator DariusTiloiu23Darius Tiloiu DariusTiloiu23 Data 26 octombrie 2026 14:25:12
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.05 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

const int MAX = 100000;
int p[MAX + 1], sz[MAX + 1];

int Find(int x) {
    while (x != p[x]) {
        x = p[x];
    }
    return x;
}

void Union(int x, int y) {
    x = Find(x);
    y = Find(y);
    if (x != y) {
        if (sz[x] > sz[y]) {
            p[y] = x;
            sz[x] += sz[y];
        } else {
            p[x] = y;
            sz[y] += sz[x];
        }
    }
}

int main() {
    ifstream fin("disjoint.in");
    ofstream fout("disjoint.out");

    int n, q;
    fin >> n >> q;

    for (int i = 1; i <= n; i++) {
        p[i] = i;
        sz[i] = 1;
    }

    for (int i = 1; i <= q; i++) {
        int op, x, y;
        fin >> op >> x >> y;
        if (op == 1) {
            Union(x, y);
        } else {
            if (Find(x) == Find(y)) {
                fout << "DA\n";
            } else {
                fout << "NU\n";
            }
        }
    }

    fin.close();
    fout.close();

    return 0;
}