Cod sursa(job #2755528)

Utilizator alexandruLAlungu alexandru alexandruLA Data 27 mai 2021 16:29:40
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.7 kb
#include <fstream>

using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int T[100005],rang[100005],m,n;
int multime (int x)
{
    if(T[x]!=x)
        T[x]=multime(T[x]);
    return T[x];
}
void reuneste(int x,int y)
{
    x=multime(x);
    y=multime(y);

    if(rang[x]<rang[y]) T[x]=y;
    else if(rang[x]>rang[y]) T[y]=x;
    else T[y]=x;
    if(rang[x]==rang[y]) rang[x]++;
}
int main()
{
    int x,y,c;
    f>>n>>m;
    for(int i=1; i<=n; i++)
        T[i]=i;
    for(int i=1; i<=m; i++)
    {
        f>>c>>x>>y;
        if(c==1) reuneste(x,y);
        else if(multime(x)==multime(y)) g<<"DA"<<endl;
        else g<<"NU"<<endl;
    }
    return 0;
}