Cod sursa(job #3363713)

Utilizator medeeavasile56@gmail.comVasile Medeea [email protected] Data 21 august 2026 17:08:08
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.09 kb
#include <fstream>
#include <vector>
#include <unordered_map>
using namespace std;
ifstream cin("disjoint.in");
ofstream cout("disjoint.out");
unordered_map<int,int> parent;
unordered_map<int,int> sz;
vector<bool> sol;
void make_group(int x)
{
    if(parent.find(x)==parent.end())
    {
        parent[x]=x;
        sz[x]=1;
    }
}
int reprezentant(int x)
{
    if(parent[x]==x) return x;
    return parent[x]=reprezentant(parent[x]);
}
bool verif(int x,int y)
{
    return reprezentant(x)==reprezentant(y);
}
void union_(int x,int y)
{
    make_group(x);make_group(y);
    int rx=reprezentant(x), ry=reprezentant(y);
    if(sz[rx]<sz[ry]) swap(rx,ry);
    parent[ry]=rx;
    sz[rx]+=sz[ry];
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++) make_group(i);
    for(int i=1;i<=m;i++)
    {
        int x,y,op; cin>>op>>x>>y;
        if(op==1)
        {
            union_(x,y);
            continue;
        }
        sol.push_back(verif(x,y));
    }
    for(auto ok:sol) if(ok) cout<<"DA"<<'\n';
    else cout<<"NU"<<'\n';
    return 0;
}