Cod sursa(job #3370485)

Utilizator creduardCremenescu Eduard-Gabriel creduard Data 26 octombrie 2026 14:23:50
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.91 kb
#include <bits/stdc++.h>
#include <fstream>
using namespace std;

const int NMAX = 1e5;
int p[NMAX + 1], sz[NMAX + 1];

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

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 (sz[x] > sz[y]) {
        p[y] = x;
        sz[x] += sz[y];
    } else {
        p[x] = y;
        sz[y] += sz[x];
    }
}

int main() {
    int n,q;
    f >> 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;
        f >> op >> x >> y;
        if (op == 1) {
            Union(x,y);
        } else {
            if (Find(x) != Find(y)) {
                g << "NU\n";
            } else {
                g << "DA\n";
            }
        }
    }
    return 0;
}