Cod sursa(job #3217482)

Utilizator sireanu_vladSireanu Vlad sireanu_vlad Data 23 martie 2024 10:53:06
Problema Paduri de multimi disjuncte Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.03 kb
#include <iostream>
using namespace std;

#define NMAX 100000

int n, m;
int rad[NMAX + 1];
int card[NMAX + 1];

void join(int x, int y) {
    card[rad[x]] += card[rad[y]];
    card[rad[y]] = 0;
    y = rad[y];
    for (int i = 1; i <= n; i++) {
        if (rad[i] == y) {
            rad[i] = rad[x];
        }
    }
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    freopen("disjoint.in", "r", stdin);
    freopen("disjoint.out", "w", stdout);

    cin >> n >> m;

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

    for (int i = 0; i < m; i += 1) {
        int cod, x, y;
        cin >> cod >> x >> y;
        if (cod == 1) {
            if (card[rad[x]] > card[rad[y]]) {
                join(x, y);
            } else {
                join(y, x);
            }
        } else {
            if (rad[x] == rad[y]) {
                cout << "DA\n";
            } else {
                cout << "NU\n";
            }
        }
    }

    return 0;
}