Pagini recente » Cod sursa (job #421110) | Cod sursa (job #2337471) | Cod sursa (job #3268859) | Cod sursa (job #840107) | Cod sursa (job #2936241)
#include <iostream>
#include<fstream>
#include<vector>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
vector<int>tata,h,dimensiune;
int n,m;
void init(){
tata.resize(n+1,0);
h.resize(n+1,0);
dimensiune.resize(n+1,0);
}
int Find(int u)
{
if (tata[u]==0)return u;
tata[u]=Find(tata[u]);
return tata[u];
}
int Union(int u , int v)
{
int repU=Find(u);
int repV=Find(v);
if(repU==repV)return 0;
if(h[repU]==h[repV])
{
tata[repV]=repU;
h[repU]++;
dimensiune[repU]+=dimensiune[repV];
return dimensiune[repU];
}
else
if(h[repU]>h[repV])
{
tata[repV]=repU;
h[repU]++;
dimensiune[repU]+=dimensiune[repV];
return dimensiune[repU];
}
else
{
tata[repU]=repV;
dimensiune[repV]+=dimensiune[repU];
return dimensiune[repV];
}
}
int main()
{
f>>n>>m;
init();
int cod,x,y;
for(int i=0;i<m;i++)
{
f>>cod>>x>>y;
if(cod==1)
{
Union(x,y);
}
else
{
if(Find(x)==Find(y))
g<<"DA"<<endl;
else
g<<"NU"<<endl;
}
}
return 0;
}