Cod sursa(job #2837435)

Utilizator andreitabaraandrei2004 andreitabara Data 22 ianuarie 2022 10:36:39
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.82 kb
#include <fstream>

using namespace std;
ifstream in ("disjoint.in");
ofstream out ("disjoint.out");
int t[100002],h[100002],n,m,x,y,c;
void unire(int x,int y)
{
    int r1=x;
    int r2=y;
    while(r1!=t[r1])r1=t[r1];
    while(r2!=t[r2])r2=t[r2];
    if(h[r1]>h[r2])
    {
        t[r2]=r1;
    }
    else
    {
        t[r1]=r2;
        if(h[r1]==h[r2])h[r2]++;
    }
}
int mult(int x,int y)
{
    int r1=x;
    int r2=y;
    while(r1!=t[r1])r1=t[r1];
    while(r2!=t[r2])r2=t[r2];
    if(r1==r2)return 1;
    return 0;
}

int main()
{
in>>n>>m;
for(int i=1;i<=n;i++){t[i]=i,h[i]=1;}
for(int i=1;i<=m;i++)
{
    in>>c>>x>>y;
    if(c==1)
    {
        unire(x,y);
    }
    else
    {
        if(mult(x,y)==1)out<<"DA"<<'\n';
        else out<<"NU"<<'\n';
    }
}

    return 0;
}