Cod sursa(job #2677005)

Utilizator robeert.77Chirica Robert robeert.77 Data 25 noiembrie 2020 17:34:03
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.89 kb
#include <bits/stdc++.h>
#define ll long long
using namespace std;
ifstream fin ("disjoint.in");
ofstream fout ("disjoint.out");
int n, m, tree[100005], level[100005];

int root(int node) {
    if (node == tree[node])
        return node;
    return root(tree[node]);
}

int main() {
    fin.tie(0);
    fout.tie(0);
    ios::sync_with_stdio(0);
    fin >> n >> m;
    for (int i = 1; i <= n; i++) {
        tree[i] = i;
        level[i] = 1;
    }
    int operation, a, b;
    for (int i = 0; i < m; i++) {
        fin >> operation >> a >> b;
        a = root(a);
        b = root(b);
        if (operation == 1) {
            if (a > b)
                swap(a, b);
            tree[a] = b;
            level[b] = max(level[a] + 1, level[b]);
        }
        else if (a == b)
            fout << "DA\n";
        else
            fout << "NU\n";
    }
    return 0;
}