Cod sursa(job #2942122)

Utilizator Nicolae11Mihaila Nicolae Nicolae11 Data 18 noiembrie 2022 23:09:56
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.08 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m,op,x,y,rang[100005],radacina[100005];

int care_radacina(int x)
{
    int rad=x;
    while(rad!=radacina[rad])
    {
        rad=radacina[rad];
    }
    while(radacina[x]!=x)
    {
        int aux=radacina[x];
        radacina[x]=rad;
        x=aux;
    }
    return rad;
}

void unire(int x, int y)
{
    if(rang[x]<rang[y])
        radacina[x]=y;
    else
        radacina[y]=x;
    if(rang[x]==rang[y])
        rang[x]++;
    return;
}
int main()
{   f>>n>>m;
    for(int i=1;i<=n;i++)
    {
        radacina[i]=i;
        rang[i]=1;
    }
    for(int i=1;i<=m;i++)
    {   f>>op>>x>>y;
        if(op==1)
        {
            if(care_radacina(x)!=care_radacina(y))
                unire(care_radacina(x),care_radacina(y));
        }
        else
        {
            if(care_radacina(x)==care_radacina(y))
                g<<"DA"<<'\n';
            else
                g<<"NU"<<'\n';
        }
    }
    return 0;
}