Cod sursa(job #2553518)

Utilizator AlexAlAxAxelAlAx AlAx AlexAlAxAxel Data 22 februarie 2020 08:51:04
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.93 kb
#include <fstream>

using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");

const int nlim=100005;
int n,m,tt[nlim],rg[nlim];

void unire(int x, int y)
{
    if(rg[x]<rg[y])
        tt[x]=y;
    if(rg[x]>rg[y])
        tt[y]=x;
    if(rg[x]==rg[y])
    {
        tt[x]=y;
        rg[y]++;
    }
}

int tatal(int x)
{
    while(tt[x]!=x)
        x=tt[x];
    return x;
}

int main()
{
    int cod,x,y;
    f>>n>>m;
    for(int i=1;i<=n;i++)
    {
        tt[i]=i;
        rg[i]=1;
    }
    for(int i=1;i<=m;i++)
    {
        f>>cod>>x>>y;
        if(cod==1)
        {
            int a=tatal(x);
            int b=tatal(y);
            if(a!=b)
                unire(x,y);
        }
        if(cod==2)
        {
            int a=tatal(x);
            int b=tatal(y);
            if(a==b) g<<"DA"<<'\n';
            else g<<"NU"<<'\n';
        }
    }
    return 0;
}