Cod sursa(job #3262096)

Utilizator bogdibogdiAndrei Bogdan bogdibogdi Data 8 decembrie 2024 18:21:43
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.99 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin ("disjoint.in");
ofstream fout ("disjoint.out");

struct DSU
{
    int n;
    vector <int> par, sz;

    DSU (int _n)
    {
        n=_n;
        par.resize(n+1);
        sz.resize(n+1);
        for(int i=1;i<=n;i++) par[i]=i;
    }

    int gasesc(int nod)
    {
        if(par[nod]==nod) return nod;
        else return (par[nod]=gasesc(par[nod]));
    }

    int update(int x, int y)
    {
        x=gasesc(x);
        y=gasesc(y);
        if(sz[x]<sz[y]) swap(x,y);
        par[y]=x;
        sz[x]+=sz[y];
    }

    bool verif(int x, int y)
    {
        return (gasesc(x)==gasesc(y));
    }
};

int main()
{
    int n,m;
    fin>>n>>m;
    DSU arb(n);
    while(m--)
    {
        int t,a,b;
        fin>>t>>a>>b;
        if(t==1) arb.update(a,b);
        else
        {
            if(arb.verif(a,b)) fout<<"DA"<<endl;
            else fout<<"NU"<<endl;
        }
    }
    return 0;
}