Cod sursa(job #3343170)

Utilizator iustin.dumiDumitrescu Iustin iustin.dumi Data 26 februarie 2026 16:26:20
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.91 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int n,t[100005],op,x,y,m,h[100005],r1,r2,aux,i;
int cautare(int x)
{ while(t[x]!=x) x=t[x];
  return x;
}
int main()
{
    fin>>n>>m;
    for(i=1;i<=n;i++) t[i]=i;
    for(i=1;i<=m;i++)
    {
        fin>>op;
        if(op==2)
           {
            fin>>x>>y;
            if(cautare(x)==cautare(y)) fout<<"DA"<<'\n';
            else fout<<"NU"<<'\n';
           }
        else
        {
          fin>>x>>y;
          r1=cautare(x);
          r2=cautare(y);
          if(h[r1]<h[r2]) t[r1]=r2;
          else t[r2]=r1;
          if(h[r1]==h[r2]) h[r1]++;
          while(x!=r1)
          {aux=t[x];
           t[x]=r1;
           x=aux;
          }
          while(y!=r2)
          {aux=t[y];
           t[y]=r2;
           y=aux;
          }
        }
    }
    return 0;
}