Pagini recente » Cod sursa (job #2379852) | Ședință 2009-11-26 | Cod sursa (job #3174865) | Cod sursa (job #2497430) | Cod sursa (job #2946546)
#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) {
if (parentVec[a] == -1) {
return a;
}
parentVec[a] = GetRoot(parentVec[a]);
return parentVec[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;
}