Cod sursa(job #2942385)

Utilizator adelinapetreAdelina Petre adelinapetre Data 19 noiembrie 2022 17:06:44
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.81 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin("disjoint.in");
ofstream cout("disjoint.out");
int tip[100005],cnt[100005];
int parinte(int u)
{
    if(u==tip[u])
        return u;
    return tip[u]=parinte(tip[u]);
}
void join(int a, int b)
{
    a=parinte(a),b=parinte(b);
    if (a!=b)
        if (cnt[a]>cnt[b])
            swap(a, b);
    tip[a]=b;
    cnt[b]+=cnt[a];
}
int main()
{
    int n,m,i,j,op,x,y;
    cin>>n>>m;
    for(i=1; i<=n; i++)
        tip[i]=i,cnt[i]=1;
    for(i=1; i<=m; i++)
    {
        cin>>op>>x>>y;
        if(op==1)
            join(x,y);
        else
        {
            if(parinte(x)==parinte(y))
                cout<<"DA";
            else
                cout<<"NU";
            cout<<'\n';
        }
    }
    return 0;
}