Cod sursa(job #1771246)

Utilizator calin1Serban Calin calin1 Data 5 octombrie 2016 13:42:19
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.37 kb
#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

int n,m;
int tata[100005],adancime[100005];

void initializare()
{
    for(int i = 1 ; i <= n ; ++i)
    {
        tata[i] = i;
    }
}

int radacina(int x)
{
    while(tata[x] != x)
    {
        x = tata[x];
    }
    return x;
}

void compresie(int x, int y)
{
    int tmp;
    while(tata[y] != y)
    {
        tmp = y;
        y = tata[y];
        tata[tmp] = x;
    }
}

void join(int x, int y)
{
    if(adancime[x] == adancime[y])
    {
       tata[radacina(y)] = radacina(x);
       adancime[x]++;
       return;
    }

    if(adancime[x] < adancime[y])
    {
       swap(x,y);
    }

    tata[radacina(y)] = radacina(x);

    compresie(x,y);
}

void citire()
{
    scanf("%d %d\n",&n,&m);

    initializare();

    int tmp,x,y;

    for(int i = 0 ; i < m ; ++i)
    {
        scanf("%d %d %d\n",&tmp,&y,&x);

        if(tmp == 1)
        {
            join(x,y);
        }
        else
        {
            if(radacina(x) == radacina(y))
            {
                printf("DA\n");
            }
            else
            {
                printf("NU\n");
            }
        }
    }
}

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

    citire();

    return 0;
}