Cod sursa(job #2973595)

Utilizator Darius1414Dobre Darius Adrian Darius1414 Data 1 februarie 2023 13:20:39
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.78 kb
#include <fstream>
using namespace std;
int n,m,c,x,y,roo[100005],len[100005];
int cauta(int a)
{
    if (roo[a]==a)
        return a;
    return roo[a]=cauta(roo[a]);
}
void merg (int a,int b)
{
    if (len[a]<len[b])
        swap(a,b);
    len[a]+=len[b];
    roo[b]=a;
}
int main()
{
    ifstream f ("disjoint.in");
    ofstream g ("disjoint.out");
    f>>n>>m;
    for (int i=1;i<=n;i++)
        roo[i]=i,len[i]=1;
    for (int i=1;i<=m;i++)
    {
        f>>c>>x>>y;
        if (c==1)
        {
            int xs=cauta(x),ys=cauta(y);
            if (xs!=ys)
                merg(xs,ys);
        }
        else
        {
            int xs=cauta(x),ys=cauta(y);
            if (xs==ys) g<<"DA"<<'\n';
            else g<<"NU"<<'\n';
        }
    }
}