Cod sursa(job #3001568)

Utilizator Luka77Anastase Luca George Luka77 Data 13 martie 2023 19:20:24
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.28 kb
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define FOR(WHATEVER) for(int i = 1; i <= WHATEVER; ++ i)

/// INPUT / OUTPUT
const string problem = "disjoint";
ifstream fin(problem + ".in");
ofstream fout(problem + ".out");

/// GLOBAL VARIABLES
const ll NMAX = 1e5+5, MOD = 1e9 + 7, INF = 1e9;
int n, q, sets[NMAX], parent[NMAX];

/// SOLUTION
inline int Find(int x)
{
    if(parent[x] == x)
        return x;
    parent[x] = Find(parent[x]);
    return parent[x];
}

inline void Union(int x, int y)
{
    int parX = Find(x), parY = Find(y);
    if(sets[parX] < sets[parY])
    {
        swap(x, y);
        swap(parX, parY);
    }
    sets[parX] += sets[parY];
    parent[parY] = parX;
}

/// READING THE INPUT
int main()
{
    ios::sync_with_stdio(false);
    fin.tie(NULL);
    fout.tie(NULL);

    fin >> n >> q;

    for(int i = 1; i <= n; ++ i)
    {
        sets[i] = 1;
        parent[i] = i;
    }

    while(q--)
    {
        int tip, x, y;
        fin >> tip >> x >> y;
        if(tip == 1)
        {
            Union(x, y);
        }
        else
        {
            if(Find(x) == Find(y))
                fout << "DA" << '\n';
            else
                fout << "NU" << '\n';
        }
    }
}