Cod sursa(job #2924527)

Utilizator Andrei_EneaAndrei Enea Andrei_Enea Data 4 octombrie 2022 11:09:31
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.22 kb
#include <bits/stdc++.h>

using namespace std;
struct DSU
{
      int N;
      vector<int>parent,sizez;
      void init(int n)
      {
            parent.resize(n+1);
            sizez.resize(n+1,1);
            for(int i=1;i<=n;i++)
            {
                  parent[i]=i;
            }
      }
      int finb(int u)
      {
            if(u==parent[u])
                  return u;
            return parent[u]=finb(parent[u]);
      }
      void unite(int u,int v)
      {
            u=finb(u);
            v=finb(v);
            if(u==v)
            {
                  return;
            }
            if(sizez[v]>sizez[u])
                  swap(u,v);
            parent[v]=u;
            sizez[u]+=sizez[v];
      }
};
DSU dsu;
int main()
{
      ifstream cin("disjoint.in");
      ofstream cout("disjoint.out");
    int n,m,a,b,c;
    cin>>n>>m;
    dsu.init(n+1);
    for(int i=1;i<=m;i++)
    {
          cin>>a>>b>>c;
          if(a==1)
          {
                dsu.unite(b,c);
          }
          else
          {
                if(dsu.finb(b)==dsu.finb(c))cout<<"DA";
                else cout<<"NU";
                cout<<'\n';
          }
    }
    return 0;
}