Cod sursa(job #3296958)

Utilizator vladsoartavlad sofronea vladsoarta Data 19 mai 2025 15:49:01
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <fstream>
#include <vector>
using namespace std;

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

int grup[100001];

int whoIsDaddy(int startnode)
{
    int current = startnode;
    vector<int>passedby;

    while(grup[current] != current)
    {
        passedby.push_back(current);
        current = grup[current];
    }

    for(auto node:passedby)
    {
        grup[node] = current;
    }

    return current;
}

void makeHimMyDaddy(int newDaddy,int DaddysNewToy){

    int root = whoIsDaddy(DaddysNewToy);
    grup[root] = newDaddy;
}


int main()
{
    int n,q,i,j;
    cin>>n>>q;
    for(i=1;i<=n;i++)
    {
        grup[i]=i;
    }
    for(j=1;j<=q;j++)
    {
        int a,b,c;
        cin>>c>>a>>b;
        if(c==1)
        {
            makeHimMyDaddy(a,b);
        }
        else
        {
            if(whoIsDaddy(a) == whoIsDaddy(b))
            {cout<<"DA"<<'\n';
            continue;
            }
            cout<<"NU"<<'\n';
        }
    }
    return 0;
}