Cod sursa(job #3248407)

Utilizator SerbanCaroleSerban Carole SerbanCarole Data 11 octombrie 2024 17:17:03
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.9 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin("disjoint.in");
ofstream cout("disjoint.out");
const int nmax = 1e5 + 1;
vector <int> m[nmax];
int sz[nmax] , r[nmax] , n , q , op , a , b;
signed main()
{
    cin >> n;
    for(int i = 1 ; i <= n ; ++i)
    {
        sz[i] = 1;
        r[i] = i;
        m[i].push_back(i);
    }
    cin >> q;
    for(int i = 1 ; i <= q ; ++i)
    {
        cin >> op >> a >> b;
        if(op == 2)
        {
            if(r[a] == r[b]) cout << "DA\n";
            else cout << "NU\n";
        }
        else
        {
            if(r[a] == r[b]) continue;
            if(sz[r[a]] < sz[r[b]]) swap(a,b);
            sz[r[a]] += sz[r[b]];
            sz[r[b]] = 0;
            int nig = r[b];
            for(auto it : m[nig]) m[r[a]].push_back(it) , r[it] = r[a];
            m[nig].clear();
        }
    }
	return 0;
}