Cod sursa(job #3128761)

Utilizator amunnumeVlad Patrascu amunnume Data 10 mai 2023 20:46:20
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.67 kb
#include <bits/stdc++.h>
#define N 100'005
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int n,m,i,op,x,y;
int c[N],r[N];
int look(int x)
{
    if(r[x]==x) return x;
    r[x]=look(r[x]);
    return r[x];
}
int unite(int x,int y)
{
    if(c[x]<c[y]) swap(x,y);
    c[x]+=c[y];
    r[y]=x;
}
int main()
{
    fin>>n>>m;
    for(i=1;i<=n;++i)
    {
        c[i]=1;
        r[i]=i;
    }
    for(int w=1;w<=m;++w)
    {
        fin>>op>>x>>y;
        if(op==1) unite(look(x),look(y));
        else
        {
            if(look(x)==look(y)) fout<<"DA\n";
            else fout<<"NU\n";
        }
    }
    return 0;
}