Cod sursa(job #2530812)

Utilizator vladadAndries Vlad Andrei vladad Data 25 ianuarie 2020 12:35:50
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.8 kb
#include<bits/stdc++.h>
#define ll long long
#pragma GCC optimize("O3")
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n, q, dad[100001], rang[100001], tip, a, b;
int Find(int x)
{
    if(dad[x]==x)
        return x;
    return dad[x]==Find(dad[x]);
}
void Union(int x, int y)
{
    if(rang[x]>rang[y])
        dad[y]=x, rang[x]+=rang[y];
    else
        dad[x]=y, rang[y]+=rang[x];
}
int main()
{
    f>>n>>q;
    for(int i=1; i<=n; i++)
        dad[i]=i, rang[i]=1;
    for(; q; q--)
    {
        f>>tip>>a>>b;
        if(tip==1)
            Union(Find(a), Find(b));
        else
        {
            if(Find(a)==Find(b))
                g<<"DA";
            else
                g<<"NU";
            g<<'\n';
        }
    }
    return 0;
}