Cod sursa(job #2977206)

Utilizator theo_aldescuTheodora Aldescu theo_aldescu Data 11 februarie 2023 09:10:55
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.6 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int t[100001],h[100001];
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];
//regula de compresie
while(x!=r)
    {int aux=t[x];
    t[x]=r;
    x=aux;
    }
return r;
}


int main()
{int n,m,i,cod,x,y;
f>>n>>m;
for(i=1;i<=n;i++)
    {t[i]=i;h[i]=1;}
for(i=1;i<=m;i++)
    {f>>cod>>x>>y;
    if(cod==1)
      unif(x,y);
    else {if(rad(x)==rad(y))g<<"DA"<<'\n';    
            else g<<"NU"<<'\n';}
    }

}