Cod sursa(job #3228391)

Utilizator catalinmarincatalinmarin catalinmarin Data 8 mai 2024 09:20:16
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.92 kb
#include <fstream>
#include <map>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int root[int(1e5) + 5];
int sz[int(1e5) + 5];
int find_root(int x){
    if (root[x] == x)
        return x;
    root[x] = find_root(root[x]);
    return root[x];
}
void join(int r1, int r2){
    if (sz[r1] < sz[r2]){
        swap(r1, r2);
    }
    root[r2] = r1;
    sz[r1] += sz[r2];
}
int main(){
    int m, n;
    fin >> n >> m;
    for (int i = 1; i <= n; i++){
        sz[i] = 1;
        root[i] = i;
    }
    for (int i = 1; i <= m; i++){
        int a, b, q;
        fin >> q >> a >> b;
        int root_1 = find_root(a);
        int root_2 = find_root(b);
        if (q == 1 && root_1 != root_2){
            join(root_1, root_2);
        } else if (q == 2){
            if (root_1 == root_2)
                fout << "DA" << '\n';
            else
                fout << "NU" << '\n';
        }
    }
    return 0;
}