Cod sursa(job #3271820)

Utilizator Lupu_Daniel_24Lupu Daniel Lupu_Daniel_24 Data 27 ianuarie 2025 14:23:31
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.82 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m,t[100001],h[100001],o,x,y,i,r1,r2;
void unif(int r1,int r2)
{
    if (h[r1]<h[r2]) t[r1]=r2;
    else t[r2]=r1;
    if (h[r1]==h[r2]) h[r1]++;
}
int rad(int x)
{
    int  r=x;
    while(t[r]!=r)
        r=t[r];
    while(x!=r)
    {
        int    aux=t[x];
        t[x]=r;
        x=aux;
    }
    return r;
}
int main()
{
    f>>n>>m;
    for (i=1; i<=n; i++)
    {
        h[i]=1;
        t[i]=i;
    }
    for (i=1; i<=m; i++)
    {
        f>>o>>x>>y;
        r1=rad(x);
        r2=rad(y);

        if (o==1)
        {

            if(r1!=r2)unif(r1,r2);
        }
        else

            if (r1==r2) g<<"DA"<<'\n';
            else g<<"NU"<<'\n';
    }

    return 0;
}