Cod sursa(job #2495576)

Utilizator stefantagaTaga Stefan stefantaga Data 19 noiembrie 2019 17:39:10
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m,i,tata[100005],x,y,tip,rang[100005];

int repr (int x)
{
    if (tata[x]!=x)
    {
        return repr(tata[x]);
    }
    return tata[x];
}
void uneste (int x,int y)
{
    x=repr(x);
    y=repr(y);
    if (rang[x]<rang[y])
    {
        tata[x]=y;
    }
    else
    if (rang[x]>rang[y])
    {
        tata[y]=x;
    }
    else
    {
        tata[y]=x;
        rang[x]++;
    }
}
int main()
{
    f>>n>>m;
    for (i=1;i<=n;i++)
    {
        tata[i]=i;
        rang[i]=1;
    }
    for (i=1;i<=m;i++)
    {
        f>>tip>>x>>y;
        if (tip==1)
        {
            if (repr(x)!=repr(y))
            {
                uneste(x,y);
            }
        }
        else
        {
            if (repr(x)==repr(y))
            {
                g<<"DA"<<'\n';
            }
            else
            {
                g<<"NU"<<'\n';
            }
        }
    }
    return 0;
}