Cod sursa(job #2227929)

Utilizator inquisitorAnders inquisitor Data 2 august 2018 11:03:17
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.4 kb
#include <stdio.h>

int parent[100001], rank[100001], p = -1, q = -1;

char buffer[1400000], outBuffer[300000];

__attribute__((always_inline)) int get_int()
{
    int number = 0;

    for(++p; buffer[p] > 47; ++p)
    {
        number = number * 10 + buffer[p] - 48;
    }

    return number;
}

__attribute__((always_inline)) int Find(int x)
{
    int root;

    for(root = x; parent[root] != root; root = parent[root]);

    return root;
}

__attribute__((always_inline)) void Union(int x, int y)
{
    parent[x] = y;
}

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

    fread(buffer, 1, 1400000, stdin);

    int N = get_int(), M = get_int();

    for(int i = 1; i <= N; ++i)
    {
        parent[i] = i;
    }

    for(int i = 1; i <= M; ++i)
    {
        int task = get_int(), x = get_int(), y = get_int();

        if(task == 2)
        {
            if(Find(x) == Find(y))
            {
                outBuffer[++q] = 'D';
                outBuffer[++q] = 'A';
                outBuffer[++q] = '\n';
            }
            else
            {
                outBuffer[++q] = 'N';
                outBuffer[++q] = 'U';
                outBuffer[++q] = '\n';
            }
        }
        else
        {
            Union(Find(x), Find(y));
        }
    }

    puts(outBuffer);

    return 0;
}