Cod sursa(job #2419698)

Utilizator VersinMadalinaVERSIN IONELA MADALINA VersinMadalina Data 9 mai 2019 11:25:38
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.15 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int tata[200005],grad[200005];
int n,m,op;
int find_father(int node)
{
    if(tata[node]!=node)
    tata[node]=find_father(tata[node]);
    return tata[node];
}
int main()
{
    int i,x,y;
   f>>n>>m;
   for(i=1;i<=n;i++)
   {
       tata[i]=i;
       grad[i]=1;
   }
   for(i=0;i<m;i++)
    {
        f>>op>>x>>y;
        if(op==2)
        {
            int fx=find_father(x);
            int fy=find_father(y);
            if(fx==fy)
              g<<"DA"<<'\n';
            else g<<"NU"<<'\n';
        }
        else
        if(op==1)
        {
            int fx=find_father(x);
            int fy=find_father(y);
            if(fx==fy)
                continue;
            if(grad[fx]<grad[fy])
            {
                tata[fx]=fy;
                grad[fy]+=grad[fx];
            }
            else
                {
                    grad[fx]+=grad[fy];
                    tata[fy]=fx;
                }

        }
    }
    f.close();
    g.close();
    return 0;
}