Cod sursa(job #2946544)

Utilizator DevCrutCorolevschi Mihai DevCrut Data 24 noiembrie 2022 23:10:58
Problema Paduri de multimi disjuncte Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.34 kb
#include <fstream>
#include <string.h>
#include <vector>
#include <cmath>
#include <queue>
#include <iostream>


using namespace std;

vector<int> parentVec;
vector<int> distVec;
int maxN, opCount, opType, a, b;

int GetRoot(int a) {
    while (parentVec[a] != -1) {
        a = parentVec[a];
    }
    return a;
}

void UnionSet(int a, int b) {
    int rootA = GetRoot(a);
    int rootB = GetRoot(b);

    int distA = distVec[rootA];
    int distB = distVec[rootB];

    if (distA < distB) {
        parentVec[rootA] = rootB;
    }
    else if(distA > distB) {
        parentVec[rootB] = rootA;
    }
    else {
        parentVec[rootB] = rootA;
        distVec[rootA] = distA + 1;
    }
}

int main() {
    ifstream myIn("disjoint.in");
    myIn >> maxN >> opCount;

    parentVec.resize(maxN, -1);
    distVec.resize(maxN, 0);

    ofstream myOut("disjoint.out");
    for (int i = 0; i < opCount; i++) {
        myIn >> opType >> a >> b;
        a -= 1; b -= 1;

        if (opType == 1) {
            // reuniune
            UnionSet(a, b);
        }
        else {
            // bool
            if (GetRoot(a) == GetRoot(b)) {
                myOut << "DA" << endl;
            }
            else {
                myOut << "NU" << endl;
            }
        }
    }
    myOut.close();
    return 0;
}