Cod sursa(job #2866942)

Utilizator radu.Radu Cristi radu. Data 10 martie 2022 09:01:44
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.96 kb
#include <iostream>
#include <fstream>
#define NMAX 100005
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int t[NMAX], r[NMAX];
int N, M;
void read() {
    fin >> N >> M;
    for (int i = 1; i <= N; ++i) {
        t[i] = i;
    }
}
int findRoot(int x) {
    if (t[x] == x) {
        return x;
    }
    t[x] = findRoot(t[x]);
    return t[x];
}
void mergeSets(int x, int y) {
    x = findRoot(x);
    y = findRoot(y);
    if (r[x] < r[y]) {
        swap(x, y);
    }
    if (r[x] == r[y]) {
        r[x]++;
    }
    t[y] = x;
}
int main()
{
    read();
    int a, b, c;
    for (int i = 0; i < M; ++i) {
        fin >> a >> b >> c;
        if (a == 2) {
            if (findRoot(b) == findRoot(c)) {
                fout << "DA\n";
            }
            else {
                fout << "NU\n";
            }
        }
        else {
            mergeSets(b, c);
        }
    }
    return 0;
}