Cod sursa(job #2951557)

Utilizator Alex18maiAlex Enache Alex18mai Data 6 decembrie 2022 19:14:11
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.49 kb
#include <fstream>
#include <vector>
#include <queue>
#include <algorithm>

using namespace std;

//ifstream cin ("input"); ofstream cout ("output");
ifstream cin ("disjoint.in"); ofstream cout ("disjoint.out");

vector<int> tata, cardinal;

int gasesteTata(int node){
    if (node != tata[node]) {
        tata[node] = gasesteTata(tata[node]);
    }
    return tata[node];
}

void uneste(int nodA, int nodB) {
    nodA = gasesteTata(nodA);
    nodB = gasesteTata(nodB);
    if (cardinal[tata[nodA]] > cardinal[tata[nodB]]) {
        // torn ce este in B in A
        cardinal[tata[nodA]] += cardinal[tata[nodB]];
        cardinal[tata[nodB]] = 0;
        tata[nodB] = tata[nodA];
    }
    else {
        // torn ce este in A in B
        cardinal[tata[nodB]] += cardinal[tata[nodA]];
        cardinal[tata[nodA]] = 0;
        tata[nodA] = tata[nodB];
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

    int n, m;
    cin>>n>>m;

    tata.resize(n+1);
    cardinal.resize(n+1);

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

    for (int i=1; i<=m; i++){
        int op, x, y;
        cin>>op>>x>>y;

        if (op == 1) { // adaug o galeata la alta
            uneste(x, y);
        }
        else{
            int tataX = gasesteTata(x), tataY = gasesteTata(y);
            if (tataX == tataY){
                cout<<"DA"<<'\n';
            }
            else{
                cout<<"NU"<<'\n';
            }
        }
    }

    return 0;
}