Cod sursa(job #1968795)

Utilizator Horia14Horia Banciu Horia14 Data 17 aprilie 2017 21:09:58
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.94 kb
#include<cstdio>
#define N_MAX 100000
using namespace std;

int p[N_MAX+1], n, m;

void Init(int x)
{
    for(int i=1; i<=x; i++)
        p[i] = i;
}

int Find(int x)
{
    int r, aux;
    r = x;
    while(p[r] != r)
        r = p[r];
    while(p[x] != r)
    {
        aux = p[x];
        p[x] = r;
        x = aux;
    }
    return r;
}

int main()
{
    int i, op, x, y, rx, ry;
    FILE *fin, *fout;
    fin = fopen("disjoint.in","r");
    fout = fopen("disjoint.out","w");
    fscanf(fin,"%d%d",&n,&m);
    Init(n);
    for(i=1; i<=m; i++)
    {
        fscanf(fin,"%d%d%d",&op,&x,&y);
        if(op == 1)
        {
            rx = Find(x);
            ry = Find(y);
            p[rx] = ry;
        }
        else
        {
            if(Find(x) == Find(y))
                fprintf(fout,"DA\n");
            else fprintf(fout,"NU\n");
        }
    }
    fclose(fin);
    fclose(fout);
    return 0;
}