Cod sursa(job #1342832)

Utilizator stefantrettTrett Stefan stefantrett Data 14 februarie 2015 16:09:19
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.64 kb
#include <iostream>
#include <stdio.h>

#define nMax 100020

int a[nMax];
int b[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; a[R] != R; R = a[R]);

    //aplic compresia drumurilor
    for (; a[x] != x;)
    {
        y = a[x];
        a[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 (b[x] > b[y])
        a[y] = x;
    else a[x] = y;

    //in caz ca rangurile erau egale atunci cresc rangul noii multimi cu 1
    if (b[x] == b[y])
        b[y]++;
}

void sol()
{
    scanf("%d %d ", &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++)
    {
        a[i] = i;
        b[i] = 1;
    }

    for (i = 1; i <= M; i++)
    {
        scanf("%d %d %d", &cd, &x, &y);

        if (cd == 2)
        {
            //verific daca radacina arborilor in care se afla x respectiv y este egala
            if (find(x) == find(y))
                printf("DA\n");
            else printf("NU\n");
        }
        else //unesc radacinile arborilor corespunzatori multimilor nodurilor x respectiv y
        {
            if (find(x) == find(y))
            {
                fprintf(stderr,"%d ", i);
                return;
            }
            unite(find(x), find(y));
        }
    }
}

int main()
{
    freopen("disjoint.in", "r", stdin);
    freopen("disjoint.out", "w", stdout);

    sol();

    return 0;
}