Cod sursa(job #2662599)

Utilizator metallidethantralayerIon Cojocaru metallidethantralayer Data 24 octombrie 2020 11:41:21
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.98 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m,V[100005];
int radacina(int x)
{

    if(x==V[x])
        return x;
    else
        return radacina(V[x]);
}
void CompresieDrum(int x,int k)
{
    for(; V[x]!=x;)
    {
        int y=V[x];
        V[x]=k;
        x=y;
    }
}
void Leaga(int x,int y)
{
    if(x>y)
        V[x]=y;
    else
        V[y]=x;
}
int main()
{
    f>>n>>m;
    for(int i=1; i<=n; i++)
        V[i]=i;
    while(m--)
    {
        int cer,x,y;
        f>>cer>>x>>y;
        assert(cer==1||cer==2);
        int x1=radacina(x);
        CompresieDrum(x,x1);
        int y1=radacina(y);
        CompresieDrum(y,y1);
        if(cer==2)
        {
            if(x1==y1)
                g<<"DA\n";
            else
                g<<"NU\n";
        }
        else
        {
            if(x1!=y1)
                Leaga(x1,y1);
        }
    }



    return 0;

}