Pagini recente » Cod sursa (job #642810) | Borderou de evaluare (job #3179968) | Cod sursa (job #2872231) | Cod sursa (job #2789507) | Cod sursa (job #3278038)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
struct dsu {
vector<int> root;
vector<int> height;
dsu(int n) {
height.resize(n+1, 0);
root.resize(n+1);
for (int i = 1; i <= n; i++) {
root[i] = i;
}
}
int findRoot(int node) {
if (node != root[node]) {
root[node] = findRoot(root[node]);
}
return root[node];
}
void unionNodes(int u, int v) {
int rootU = findRoot(u);
int rootV = findRoot(v);
if(rootU == rootV) {
return;
}
if(height[rootU] < height[rootV]) {
root[rootU] = rootV;
}
else if(height[rootU] > height[rootV]) {
root[rootV] = rootU;
}
else {
root[rootU] = rootV;
height[rootV]++;
}
}
bool areConnected(int u, int v) {
return findRoot(u) == findRoot(v);
}
};
int main() {
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int nodes, instr;
fin >> nodes >> instr;
auto myDsu = dsu(nodes);
for (int i = 0; i < instr; i++) {
int code, u, v;
fin >> code >> u >> v;
if (code == 1) {
myDsu.unionNodes(u, v);
}
else {
if (myDsu.areConnected(u, v)) {
fout << "DA" << endl;
}
else {
fout << "NU" << endl;
}
}
}
fin.close();
fout.close();
return 0;
}