Cod sursa(job #2681316)

Utilizator monica_LMonica monica_L Data 5 decembrie 2020 11:33:48
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.05 kb
#include <cstdio>
using namespace std;
int n,m,t[100001],h[100001],o,x,y,a,i,r1,r2,c;

int main()
{FILE *f1 = fopen("disjoint.in","r");
 FILE *f2 = fopen("disjoint.out","w");
 fscanf(f1,"%d%d",&n,&m);

 for (i=1;i<=n;i++) {h[i]=1;t[i]=i;}

 for (i=1;i<=m;i++)
          {fscanf(f1,"%d%d%d",&o,&x,&y);

	r1=x;r2=y;
	while (r1!=t[r1]) r1=t[r1];  // r1 = radacina lui x
	while (r2!=t[r2]) r2=t[r2];  // r2 = radacina lui y

	if (o==1)

{ //  euristica 1 - regula de ponderare
  // arborele de inaltime mai mica devine subarbore al celui de inaltime mai mare
  if (h[r1]>h[r2]) t[r2]=r1;
		  else { t[r1]=r2;
            if (h[r1]==h[r2]) h[c]++;
           }
}

	else
{ if (r1==r2) fprintf(f2,"DA\n");
		 else fprintf(f2,"NU\n");
		 //  euristica 2 - regula compresie
		 // se leaga toate nodurile de pe drumul de la x la radacina direct de radacina			 while (x!=r1) {a=t[x];t[x]=r1;x=a;}
		 // se leaga toate nodurile de pe drumul de la y la radacina direct de radacina		             while (y!=r2) {a=t[y];t[y]=r2;y=a;}
		}
	}
return 0;
}