Cod sursa(job #2753338)

Utilizator VanillaSoltan Marian Vanilla Data 22 mai 2021 14:42:20
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.24 kb
#include <bits/stdc++.h>
#include <fstream>
using namespace std;
typedef long long int64;
typedef vector<int> vec;
typedef vector<int64> vec64;
    string __fname = "disjoint";
    ifstream in (__fname + ".in");
    ofstream out (__fname + ".out");
    #define cin in
    #define cout out
#define ss cout << " ";
#define nn cout << "\n";
#define ct(x) cout << x;
#define cts(x) cout << x << " ";
#define ctn(x) cout << x << "\n";
#define db(x) cout << "> " << #x << ": " << x << "\n";
#define qr queries();
void solve(int);
void YN(bool b){if (b){ctn("YES");}else{ctn ("NO");}};
void yn(bool b){if (b){ctn("Yes");}else{ctn ("No");}};
void queries(){int n;cin >> n;for (int i = 1; i <= n; i++) solve(i);}
int64 ceildiv(int64 a, int64 b) {return a / b + !!(a % b);}
// // // // // // // // // // // // // // // // // // // // // // 
/*                  TEMPLATE - VANILLA                         */
// // // // // // // // // // // // // // // // // // // // // //
const int maxn = 200200;
const int64 mod = 1000000007;
const double pi = 3.14159265359;
const int ddx[] = {-1, -1, 0, 1, 1, 1, 0, -1};
const int ddy[] = {0, 1, 1, 1, 0, -1, -1, -1};
const int dx[] = {-1, 0, 1, 0};
const int dy[] = {0, 1, 0, -1};
 
void solve(int id){

    return;
}

template <class T> class dsu {
    private:
        vector <T> parent;

    public:
        dsu(int size) {
            parent.resize(size);
        }
        
        void make_set(T val) {
            parent[val] = val;
        }

        void union_sets(T a, T b) {
            a = find_set(a);
            b = find_set(b);
            if (a != b) parent[b] = a;
        }

        T find_set (T node) {
            if (node == parent[node]) return node;
            return parent[node] = find_set(parent[node]);
        }
};
 
int main(){
    ios_base::sync_with_stdio(0);cin.tie(0);
    int n,k;
    cin >> n >> k;
    dsu <int> dsu (n + 1);
    for (int i = 1; i<= n; i++){
        dsu.make_set(i);
    }
    while (k--) {
        int t,x,y;
        cin >> t >> x >> y;
        if (t == 1 ) {
            dsu.union_sets(x, y);
        }
        else {
            if (dsu.find_set(x) == dsu.find_set(y)) {
                ctn("DA");
            }
            else{
                ctn("NU");
            }
        }
    }
    return 0;
}