Cod sursa(job #2076387)

Utilizator dariusdariusMarian Darius dariusdarius Data 26 noiembrie 2017 15:10:29
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.25 kb
#include <algorithm>
#include <fstream>
#include <iostream>
#include <vector>
using namespace std;

class DisjointDataSets {
private:
    vector<int> t, h;

    int Root(int x) {
        if (x == t[x]) {
            return x;
        }
        t[x] = Root(t[x]);
        return t[x];
    }

public:
    DisjointDataSets() = default;
    DisjointDataSets(int n) {
        h.assign(n + 1, 1);
        t.assign(n + 1, 0);
        for (int i = 1; i <= n; ++ i) {
            t[i] = i;
        }
    }

    void Unite(int x, int y) {
        x = Root(x);
        y = Root(y);
        if (h[x] <= h[y]) {
            t[x] = y;
            if (h[x] == h[y]) {
                ++ h[y];
            }
        } else {
            t[y] = x;
        }
    }

    bool SameSet(int x, int y) {
        return Root(x) == Root(y);
    }
};

int main() {
    ifstream cin("disjoint.in");
    ofstream cout("disjoint.out");
    int n, m;
    cin >> n >> m;
    DisjointDataSets d(n);
    for (int i = 1; i <= m; ++ i) {
        int type, x, y;
        cin >> type >> x >> y;
        if (type == 1) {
            d.Unite(x, y);
        } else {
            cout << (d.SameSet(x, y) ? "DA" : "NU") << "\n";
        }
    }
    return 0;
}