Cod sursa(job #2806016)

Utilizator Cosmin_117Gaicu Cosmin-Valentin Cosmin_117 Data 22 noiembrie 2021 11:38:28
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <fstream>
#define DIM 100005

using namespace std;

/// midifica numele fisierelor

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

int n, m, t[DIM], nrNod[DIM];

void init () {
    for (int i = 1; i <= n; i++)
        t[i] = i, nrNod[i] = 1;;
}

int radacina(int x) {
    if (x != t[x])
        return t[x] = radacina(t[x]);
    else
        return x;
}

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

    init();
    for (; m--; ) {
        int cod, x, y;
        cin >> cod >> x >> y;

        int radx = radacina(x);
        int rady = radacina(y);

        if (cod == 1) {
            if (nrNod[radx] > nrNod[rady]) {
                nrNod[radx] += nrNod[rady];
                t[rady] = radx;
            }
            else {
                nrNod[rady] += nrNod[radx];
                t[radx] = rady;
            }
        }
        else {
            if (radx == rady)
                cout << "DA\n";
            else
                cout << "NU\n";
        }
    }

    return 0;
}