Cod sursa(job #2138807)

Utilizator sulzandreiandrei sulzandrei Data 21 februarie 2018 21:26:51
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.96 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream in("disjoint.in");
ofstream out("disjoint.out");
#define maxn 100010
int p[maxn];
int rang[maxn]{0};
int gasesteMultime(int x)
{
    if(x == p[x])
        return x;
    else
        x = gasesteMultime(p[x]);
}
void reuneste(int x, int y)
{
    int px = gasesteMultime(x);
    int py = gasesteMultime(y);
    if(rang[px]> rang[py])
    {
        p[y] =px;
        rang[py]++;
    }
    else
    {
        p[x] = py;
        rang[px]++;
    }

}
int main()
{
    int n,m,op,x,y;
    in >> n >> m;
    for(int i = 1 ; i<= n ; i++)
        p[i] = i;
    for(int i = 0 ; i <m ; i++)
    {
        in >> op >> x >> y;
        if(op == 1)
            reuneste(x,y);
        else
        {
            if(gasesteMultime(x) == gasesteMultime(y))
                out<<"DA\n";
            else
                out<<"NU\n";
        }
    }
}