Cod sursa(job #2681320)

Utilizator monica_LMonica monica_L Data 5 decembrie 2020 11:38:19
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.31 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[r2]++;
            }
        }

        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;
}