Cod sursa(job #2842316)

Utilizator TiberiwTiberiu Amarie Tiberiw Data 31 ianuarie 2022 16:04:50
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.85 kb
#include <bits/stdc++.h>
#define Dmax 100005
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");

int T[Dmax],R[Dmax];

int root(int x)
{
    if(T[x]==x)
        return x;
    else
    {
        T[x]=root(T[x]);
        return T[x];
    }
}

void Union(int x, int y)
{
    if(R[x] > R[y])
        T[y] = x;
    else
        T[x] = y;
    if(R[x]==R[y])
        R[y]++;
}

int main()
{
    int n,m;
    f>>n>>m;
    for(int i = 1; i <= n; i++)
        T[i]=i;
    for(int i = 1; i <= m; i++)
    {
        int p,x,y;
        f>>p>>x>>y;
        if(p==1)
        {
            if(root(x)!=root(y))
                Union(root(x),root(y));
        }
        else
            if(root(x)==root(y))
                g<<"DA\n";
                else
                g<<"NU\n";
    }



    return 0;
}