Cod sursa(job #2348310)

Utilizator fremenFremen fremen Data 19 februarie 2019 16:40:40
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <fstream>
#include <cstring>
#include <algorithm>
using namespace std;

ifstream fin("disjoint.in");
ofstream fout("disjoint.out");

const int MAXN = 100005;
int n, m;
int r[MAXN], v[MAXN];

void unite(int x, int y) {

    if (r[x] >= r[y]) {
        v[y] = v[x];
    }
    else {
        v[x] = v[y];
    }
    if (r[x] == r[y]) {
        r[x]++;
    }
}

int find(int x) {
    int t;
    for (t = x; t != v[t]; t = v[t]);
    while (x != v[x]) {
        int y = v[x];
        v[x] = t;
        x = y;
    }
    return t;
}

int main() {

    fin >> n >> m;
    for (int i = 1; i <= n; ++i) {
        r[i] = 1;
        v[i] = i;
    }

    for (int i = 1; i <= m; ++i) {
        int a, b, c;
        fin >> a >> b >> c;
        if (a == 1) {
            unite(find(b), find(c));
        }
        else {
            if (find(b) == find(c)) {
                fout << "DA\n";
            }
            else {
                fout << "NU\n";
            }
        }
    }

    fout.close();
    return 0;
    }