Cod sursa(job #2535001)

Utilizator Mihai_EduardMihai Eduard Mihai_Eduard Data 31 ianuarie 2020 11:53:33
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int n, m, rang[100000], v[100000];
int radacina(int x)
{
    int i=x, j;
    while(v[i]!=i)
    {
        i=v[i];
    }
    while(v[x]!=x)
    {
        j=v[x];
        v[x]=i;
        x=j;
    }
    return i;
}
void unire(int x, int y)
{
    if(rang[x]<rang[y])
    {
        v[x]=y;
    }
    else
    {
        v[y]=x;
    }
    if(rang[x]==rang[y])
        rang[x]++;
}
int main()
{
    int x, y, cod;
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        rang[i]=1;
        v[i]=i;
    }
    for(int i=1;i<=m;i++)
    {
        fin>>cod>>x>>y;
        if(cod==1)
        {
            if(radacina(x)!=radacina(y))
            unire(radacina(x),radacina(y));
        }
        else
        {
            if(radacina(x)==radacina(y))
                fout<<"DA"<<endl;
            else
                fout<<"NU"<<endl;
        }
    }
}