Cod sursa(job #3242894)

Utilizator nistor_dora_valentinaNistor Dora Valentina nistor_dora_valentina Data 14 septembrie 2024 14:50:43
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 fin("disjoint.in");
ofstream fout("disjoint.out");
int n, m, i, j, cod, x, y, parent[100005], sze[100005];
void make_set(int nod)
{
    parent[nod]=nod;
    sze[nod]=1;
}
int find_set(int nod)
{
   if(nod==parent[nod])
        return nod;
   parent[nod]=find_set(parent[nod]);
   return parent[nod];
}
void union_set(int a, int b)
{
    a=find_set(a);
    b=find_set(b);
    if(a!=b)
    {
        if(sze[a]<sze[b])
            swap(a, b);
        parent[b]=a;
        sze[a]+=sze[b];
    }
}
int main()
{
    fin>>n>>m;
    for(i=1; i<=n; i++)
        make_set(i);
    for(i=1; i<=m; i++)
    {
        fin>>cod>>x>>y;
        if(cod==1)
            union_set(x, y);
        else
        {
            if(find_set(x)!=find_set(y))
                fout<<"NU"<<'\n';
            else
                fout<<"DA"<<'\n';
        }
    }
    return 0;
}