Pagini recente » Cod sursa (job #1117325) | Cod sursa (job #403024) | Cod sursa (job #1100028) | Cod sursa (job #559049) | Cod sursa (job #2960557)
#include<iostream>
#include<fstream>
#include<vector>
using namespace std;
struct dsu{
vector<int> sizes,parent;
void init(int n)
{
parent.resize(n+2);
sizes.resize(n+2,1);
for(int i=1;i<=n;i++)
parent[i]=i;
}
int master(int u)
{
if(parent[u]==u)
return u;
return parent[u]=master(parent[u]);
}
void unite(int u,int v)
{
u=master(u);
v=master(v);
if(u==v)
return;
if(sizes[u]<sizes[v])
swap(u,v);
parent[v]=u;
sizes[u]+=sizes[v];
}
};
int main()
{
ifstream f("disjoint.in");
ofstream g("disjoint.out");
dsu DSU;
int n,q;
f>>n>>q;
DSU.init(n);
for(int i=1;i<=q;i++)
{
int x,y,z;
f>>x>>y>>z;
if(x==1)
{
DSU.unite(y,z);
}
if(x==2)
{
if(DSU.master(y)==DSU.master(z))
g<<"DA"<<endl;
else
g<<"NU"<<endl;
}
}
}