Cod sursa(job #2482881)

Utilizator SergiuS3003Sergiu Stancu Nicolae SergiuS3003 Data 28 octombrie 2019 23:20:16
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.44 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f ( "disjoint.in" );
ofstream g ( "disjoint.out" );
const int NMAX = 100020;
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()
{
    f >> N >> M;
    int i, x, y, cd;

    //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++ )
    {
        f >> cd >> x >> y;

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

    return 0;
}