Cod sursa(job #2198305)

Utilizator codrin18Diac Eugen Codrin codrin18 Data 24 aprilie 2018 10:34:25
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.58 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cmath>
#include <algorithm>
#define NMAX 100020

using namespace std;
int TT[NMAX], RG[NMAX];
int N, M;
int Find(int x)
{
     int R, y;

    ///merg in sus pe arbore pana gasesc un nod care pointeaza catre el insusi
    for (R = x; TT[R] != R; R = TT[R]);

    ///aplic compresia drumurilor
    for (; TT[x] != x;)
    {
        y = TT[x];
        TT[x] = R;
        x = y;
    }
    return R;
}
void unite(int x, int y)
{
    ///unesc multimea cu rang mai mic de cea cu rang mai mare
    if (RG[x] > RG[y])
        TT[y] = x;
            else TT[x] = y;

    ///in caz ca rangurile erau egale atunci cresc rangul noii multimi cu 1
    if (RG[x] == RG[y]) RG[y]++;
}
int main()
{
    ifstream in("disjoint.in");
    ofstream out("disjoint.out");
    in >>N>>M;
    int x,y,cd,i;
    //initial fiecare nod pointeaza catre el insusi iar rangul fiecarui arbore este 1
    for (i = 1; i <= N; i++)
    {
        TT[i] = i;
        RG[i] = 1;
    }

    for (i = 1; i <= M; i++)
    {
        in >>cd>>x>>y;

        if (cd == 2){
            ///verific daca radacina arborilor in care se afla x respectiv y este egala
            if (Find(x) == Find(y)) out<<"DA"<<"\n";
                else out <<"NU"<<"\n";
        }
            else ///unesc radacinile arborilor corespunzatori multimilor nodurilor x respectiv y
                {
                    if (Find(x) == Find(y))  {out <<i;return 0;}
                    unite(Find(x), Find(y));
                }
    }
    return 0;
}