Cod sursa(job #2935664)

Utilizator stefan24Sandor Stefan stefan24 Data 7 noiembrie 2022 11:44:14
Problema Paduri de multimi disjuncte Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.84 kb
#include <bits/stdc++.h>
using namespace std;
#define nmax 100000
ifstream f("disjoint.in");
ofstream g("disjoint.out");

int t[nmax],rang[nmax];

int radacina(int k)
{
    if(t[k]==0)return k;
    else{
        int x=radacina(t[k]);
        t[k]=x;
        return x;
    }
}

void unire(int a,int b)
{
    int ra=radacina(a);
    int rb=radacina(b);
    if(ra!=rb){
        if(rang[ra]>rang[rb])t[rb]=ra;
        else{
            t[ra]=rb;
            if(rang[ra]==rang[rb])rang[rb]++;
        }
    }
}

int main()
{
    int n,m;
    f>>n;
    f>>m;
    for(int i=1; i<=m; ++i)
    {
        int cer,x,y;
        f>>cer>>x>>y;
        if(cer==1)unire(x,y);
        else{
            int a=radacina(x);
            int b=radacina(y);
            if(a==b)g<<"DA\n";
            else g<<"NU\n";
        }
    }
}