Pagini recente » Cod sursa (job #2670755) | Cod sursa (job #2716502) | Cod sursa (job #1249134) | Cod sursa (job #2261541) | Cod sursa (job #2944991)
#include <bits/stdc++.h>
using namespace std;
class UnionFind{
int NrNodes;
int cod, X, Y;
vector<int> Heights, Fathers;
public:
// Exclusiv pentru problema
void Solve();
// Functii union find
void Init();
int Find(int);
void Union(int,int);
};
void UnionFind::Init() {
Heights.push_back(0);
Fathers.push_back(0);
}
int UnionFind::Find(int Node) {
if(Fathers[Node] == 0)
return Node;
return Fathers[Node] = Find(Fathers[Node]);
}
void UnionFind::Union(int NodeX, int NodeY) {
int RootX = Find(NodeX), RootY = Find(NodeY);
if(Heights[RootX] < Heights[RootY]){
Fathers[RootX] = RootY;
}
else if(Heights[RootX] > Heights[RootY]){
Fathers[RootY] = RootX;
} else{
Fathers[RootY] = RootX;
Heights[RootX]++;
}
}
void UnionFind::Solve() {
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
fin >> NrNodes;
for(int i = 0; i <= NrNodes; i++){
Init();
}
int M;
fin >> M;
for(int i = 0; i < M; i++){
fin >> cod >> X >> Y;
if(cod == 1){
Union(X,Y);
} else{
if( Find(X) == Find(Y))
fout << "DA" << endl;
else
fout << "NU" << endl;
}
}
fin.close();
fout.close();
}
int main() {
UnionFind multime;
multime.Solve();
return 0;
}