Cod sursa(job #2511088)

Utilizator BogdanRuleaBogdan Rulea BogdanRulea Data 18 decembrie 2019 08:24:34
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.92 kb
#include <fstream>

using namespace std;
ifstream cin("disjoint.in");
ofstream cout("disjoint.out");
const int NMAX=100005;
int n,t[NMAX],rg[NMAX],m,task,x,y;
int find_(int nod)
{
    int r,y;
    r=nod;
    while(nod!=t[nod])
        nod=t[nod];
    for(; r!=t[r];)
    {
        y=t[r];
        t[r]=nod;
        r=y;
    }
    return nod;
}
void union_(int x,int y)
{
    if(rg[x]>rg[y])
        t[y]=x;
    else
    {
        t[x]=y;
        if(rg[x]==rg[y])
            rg[y]++;
    }
}
int main()
{
    cin>>n;
    cin>>m;
    for(int i=1; i<=n; i++)
    {
        t[i]=i;
        rg[i]=1;
    }
    for(int i=1; i<=m; i++)
    {
        cin>>task>>x>>y;
        int tx=find_(x);
        int ty=find_(y);
        if(task==1)
        {

            if(tx!=ty)
                union_(tx,ty);
        }
        else
            cout<<((tx!=ty)?"NU":"DA")<<endl;
    }
    return 0;
}