Cod sursa(job #1660733)

Utilizator Kzsolty96SAPIENTIA OSZTIAN DANIEL KUCSVAN Kzsolty96 Data 23 martie 2016 13:13:42
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.69 kb
/*#include <stdio.h>
#include <stdlib.h>

int find(int *, int);
void unite(int *, int *, int, int);

int main(){
    int n, m, c, x, y, *t, *r, i;
    freopen("disjoint.in","r", stdin);
    freopen("disjoint.out","w", stdout);
    scanf("%i%i",&n,&m);
    t = (int *) malloc((n + 1) * sizeof(int));
    r = (int *) malloc((n + 1) * sizeof(int));
    for(i = 1; i <= n; ++i){
        t[i] = i;
        r[i] = 1;
    }
    while(m){
        scanf("%i %i %i", &c, &x, &y);
        if(c == 2){
            if(find(t, x) == find(t, y)){
                printf("DA\n");
            }
            else{
                printf("NU\n");
            }
        }
        else{
            unite(t, r, x, y);
        }
        --m;
    }
    return 0;
}

int find(int *t, int x){
    int tx = x, y;
    while(t[tx] != tx){
        tx = t[tx];
    }
    while(t[x] != x){
        y = t[x];
        t[x] = tx;
        x = y;
    }
    return tx;
}

void unite(int *t, int *r, int x, int y){
    if(r[x] < r[y]){
        t[x] = y;
    }
    else{
        t[y] = x;
    }
    if(r[x] == r[y]){
        r[y]++;
    }

}
*/
#include <stdio.h>

#define NMAX 100020

int TT[NMAX], RG[NMAX];
int N, M;

int find(int x)
{
    int R, y;

    //merg in sus pe arbore pana gasesc un nod care pointeaza catre el insusi
    for (R = x; TT[R] != R; R = TT[R]);

    //aplic compresia drumurilor
    for (; TT[x] != x;)
    {
        y = TT[x];
        TT[x] = R;
        x = y;
    }
    return R;
}

void unite(int x, int y)
{
    //unesc multimea cu rang mai mic de cea cu rang mai mare
    if (RG[x] > RG[y])
        TT[y] = x;
            else TT[x] = y;

    //in caz ca rangurile erau egale atunci cresc rangul noii multimi cu 1
    if (RG[x] == RG[y]) RG[y]++;
}

int main()
{
    freopen("disjoint.in", "r", stdin);
    freopen("disjoint.out", "w", stdout);

    scanf("%d %d ", &N, &M);

    int i, x, y, cd;

    //initial fiecare nod pointeaza catre el insusi iar rangul fiecarui arbore este 1
    for (i = 1; i <= N; i++)
    {
        TT[i] = i;
        RG[i] = 1;
    }

    for (i = 1; i <= M; i++)
    {
        scanf("%d %d %d", &cd, &x, &y);

        if (cd == 2){
            //verific daca radacina arborilor in care se afla x respectiv y este egala
            if (find(x) == find(y)) printf("DA\n");
                else printf("NU\n");
        }
            else //unesc radacinile arborilor corespunzatori multimilor nodurilor x respectiv y
                {
                    if (find(x) == find(y))  {fprintf(stderr,"%d ", i);return 0;}
                    unite(find(x), find(y));
                }
    }

    return 0;
}