Cod sursa(job #3336117)

Utilizator alexandra_popaa13Popa Alexandra alexandra_popaa13 Data 24 ianuarie 2026 11:21:38
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.35 kb
#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]++;
}