Cod sursa(job #2912415)

Utilizator Giurgea3000Giurgea Giurgea3000 Data 8 iulie 2022 12:16:40
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.93 kb
#include <fstream>

using namespace std;

ifstream cin("disjoint.in");
ofstream cout("disjoint.out");

int ma[100005];
//int ma[100];

int functie(int x)
{
    if(ma[x]==x)
        return x;
    ma[x] = functie(ma[x]);
    return ma[x];
}

void unire(int a, int b)
{
    int sefa = functie(a);
    int sefb = functie(b);
    ma[sefb] = sefa;
}

int main()
{
    int n,m,c=0,x=0,y=0;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        ma[i]=i;
    }
    for(int i=1;i<=m;i++)
    {
        cin>>c>>x>>y;
        if(x>y)
        {
            int aux=x;
            x=y;
            y=aux;
        }
        if(c==1)
        {
            unire(x,y);
        }
        else
        {
            int a=functie(x);
            int b=functie(y);
            if(ma[a]==ma[b])
            {
                cout<<"DA"<<"\n";
            }
            else
            {
                cout<<"NU"<<"\n";
            }
        }
    }
    return 0;
}