Cod sursa(job #2700997)

Utilizator dolhasca001Birleanu Matei dolhasca001 Data 29 ianuarie 2021 15:38:20
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.74 kb
#include <fstream>

using namespace std;
ifstream in("disjoint.in");
ofstream out("disjoint.out");
int sef[100000],n,m;
int sef_suprem(int x)
{
    if(x==sef[x])
        return x;

        return sef[x]=sef_suprem(sef[x]);
}
void unire(int x,int y)
{
    int a,b;
    a=sef_suprem(x);
    b=sef_suprem(y);
    sef[a]=b;
}



int main()
{
    int x,y,z;
    in>>n>>m;
    for(int i=1;i<=n;i++)
        sef[i]=i;
    for(int i=1;i<=m;i++)
    {
        in>>x>>y>>z;
        if(z==1)
        {
            unire(x,y);
        }
        else
        {
            if(sef_suprem(x)==sef_suprem(y))
                out<<"DA"<<endl;
            else
                out<<"NU"<<endl;
        }
    }

    return 0;
}