Cod sursa(job #2312044)

Utilizator claudiu.gatinaFMI Claudiu Gatina claudiu.gatina Data 4 ianuarie 2019 00:52:40
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb
#include <iostream>
#include <map>
#include <vector>
#include <algorithm>
#include <cstdio>

using namespace std;

class DSet
{
public:
    vector<int> repr;
    void s_union(int x, int y)
    {
        if(repr[x] == repr[y])
            return;
        repr[repr[y]] = repr[x];
    }

    bool disjoint(int a, int b)
    {
        return findSet(a) != findSet(b);
    }

    int findSet(int x)
    {
        int chain = x;
        while(repr[x] != x)
        {
            x = repr[x];
        }
        while(repr[chain] != x)
        {
            int aux = repr[chain];
            repr[chain] = x;
            chain = aux;
        }
        return x;
    }

    DSet(int n)
    {
        for(int i = 0; i <= n; ++i)
            repr.push_back(i);
    }
};

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

    int n, m;
    scanf("%d", &n);
    scanf("%d", &m);
    DSet dSet(n);
    for(int i = 0; i < m; ++i)
    {
        int op, x, y;
        scanf("%d", &op);
        scanf("%d", &x);
        scanf("%d", &y);
        if(op == 1)
        {
            dSet.s_union(x, y);
        }
        else
        {
            if(dSet.disjoint(x, y))
                printf("NU\n");
            else
                printf("DA\n");
        }
    }

    return 0;
}