Cod sursa(job #2094640)

Utilizator alexandra_paticaAndreea Alexandra Patica alexandra_patica Data 26 decembrie 2017 12:42:46
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.85 kb
#include <cstdio>
#include <cstring>
using namespace std;

int n, m, i, t[100002], p[100002], cod, x, y, rx, ry;

int root (int x)
{
    while (t[x]!=x) x=t[x];
    return x;
}
void uni (int x, int y)
{
    if (p[x]<p[y]) t[x]=y;
    if (p[x]>p[y]) t[y]=x;
    if (p[x]==p[y]){
        t[y]=x;
        p[x]++;
    }
}
int main ()
{
    freopen("disjoint.in", "r", stdin);
    freopen("disjoint.out", "w", stdout);

    scanf("%d%d", &n, &m);
    for (i=1; i<=n; i++) t[i]=i;
    for (i=1; i<=m; i++){
        scanf("%d%d%d", &cod, &x, &y);
        if (cod==1){
            rx=root(x);
            ry=root(y);
            if (rx!=ry) uni(rx, ry);
        }
        else{
            rx=root(x);
            ry=root(y);
            if (rx==ry) printf("DA\n");
            else printf("NU\n");
        }
    }
    return 0;
}