Cod sursa(job #2936241)

Utilizator antonia2003antonia oancea antonia2003 Data 8 noiembrie 2022 13:32:59
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb
#include <iostream>
#include<fstream>
#include<vector>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
vector<int>tata,h,dimensiune;
int n,m;
void init(){
    tata.resize(n+1,0);
    h.resize(n+1,0);
    dimensiune.resize(n+1,0);
}
int Find(int u)
{
    if (tata[u]==0)return u;
    tata[u]=Find(tata[u]);
    return tata[u];
}
int Union(int u , int v)
{
    int repU=Find(u);
    int repV=Find(v);
    if(repU==repV)return 0;
    if(h[repU]==h[repV])
    {
        tata[repV]=repU;
        h[repU]++;
        dimensiune[repU]+=dimensiune[repV];
        return dimensiune[repU];
    }
    else
        if(h[repU]>h[repV])
            {
                tata[repV]=repU;
                h[repU]++;
                dimensiune[repU]+=dimensiune[repV];
                return dimensiune[repU];
            }
    else
         {
             tata[repU]=repV;
             dimensiune[repV]+=dimensiune[repU];
             return dimensiune[repV];
        }

}
int main()
{
    f>>n>>m;
    init();
    int cod,x,y;
    for(int i=0;i<m;i++)
    {
        f>>cod>>x>>y;
        if(cod==1)
        {
            Union(x,y);
        }
        else
        {
            if(Find(x)==Find(y))
                g<<"DA"<<endl;
            else
                g<<"NU"<<endl;
        }
    }
    return 0;
}