Cod sursa(job #2917009)

Utilizator vlad_lupuletiLupuleti Vlad Calin vlad_lupuleti Data 2 august 2022 17:45:23
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.96 kb
#include <fstream>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int ra[100003],h[100003],n,m,c,x,y;
int findr(int x)
{
    int r=x;
    while(ra[r]!=r)
    {
        r=ra[r];
    }
    int i=x;
    while(i!=r)
    {
        int j=ra[i];
        ra[i]=r;
        i=j;
    }
    return r;
}
void com(int a,int b)
{
    if(h[a]==h[b])
    {
        h[a]++;
        ra[b]=a;
    }
    else
    {
        if(h[a]>h[b])
        {
            ra[b]=a;
        }
        else
        {
            ra[a]=b;
        }
    }
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        h[i]=1;
        ra[i]=i;
    }
    for(int i=1;i<=m;i++)
    {
        fin>>c>>x>>y;
        if(c==1)
        {
            com(findr(x),findr(y));
        }
        if(c==2)
        {
            if(findr(x)==findr(y)) fout<<"DA\n";
            else fout<<"NU\n";
        }
    }
    return 0;
}