Cod sursa(job #2858906)

Utilizator betybety bety bety Data 28 februarie 2022 16:16:01
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.86 kb
#include <bits/stdc++.h>
using namespace std;
ifstream in("disjoint.in");
ofstream out("disjoint.out");
const int lim=1e5+4;
int n,q,t,x,y;
int link[lim];
int renk[lim];
int dim[lim];
int tata(int x)
{
    int cpy=x,aux;
    while(x!=link[x]) x=link[x];
    while(cpy!=link[cpy]) aux=cpy,cpy=link[cpy],link[aux]=x;
    return x;
}
void unite(int x,int y)
{
    x=tata(x);
    y=tata(y);
    if(x==y)
        return ;
    if(renk[x]<renk[y])
        swap(x,y);
    dim[x]+=dim[y];
    link[y]=x;
    if(renk[x]==renk[y])
        ++renk[x];
}
int main()
{
    in>>n>>q;
    for(int i=1;i<=n;++i)
        link[i]=i,
        renk[i]=1,
        dim[i]=1;
    while(q--)
    {
        in>>t>>x>>y;
        if(t==1) unite(x,y);
        else if(tata(x)==tata(y))
            out<<"DA" "\n";
        else out<<"NU" "\n";
    }
    return 0;
}