Pagini recente » Cod sursa (job #3357958) | Cod sursa (job #3299734) | Cod sursa (job #2829575) | Cod sursa (job #555730) | Cod sursa (job #3336117)
#include <fstream>
#define NMAX 100002
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int n,m;
int t[NMAX],rg[NMAX];
int gaseste(int x);
void uneste(int x,int y);
int main()
{
int i,x,y,op;
fin>>n>>m;
for(i=1;i<=n;i++){t[i]=i; rg[i]=1;}
for(i=1;i<=m;i++)
{
fin>>op>>x>>y;
if(op==2)//verificam daca x,y sunt in aceeasi multime
{
//verificam daca radacina arborilor in care apare x este egala cu
//radacina arborilor in care apare y
if(gaseste(x)==gaseste(y))
fout<<"DA"<<'\n';
else fout<<"NU"<<'\n';
}
else //unesc radacinile arborilor corespunzatori multimilor x si y
{
if(gaseste(x)!=gaseste(y)) uneste(gaseste(x),gaseste(y));
}
}
return 0;
}
int gaseste(int x)
{
int R, y;
//merg in sus pe arbore pana gasesc un nod care pointeaza catre el insusi
for (R=x;t[R]!=R;R=t[R]);
//aplic compresia drumurilor
while (t[x]!=x)
{
y=t[x];
t[x]=R;
x=y;
}
return R;
}
void uneste(int x,int y)
{
//unesc multimea cu rang mai mic de cea cu rang mai mare
if (rg[x]>rg[y])
t[y]=x;
else t[x]=y;
//in caz ca rangurile erau egale atunci cresc rangul noii multimi cu 1
if (rg[x]==rg[y]) rg[y]++;
}