Cod sursa(job #2075197)

Utilizator nerelog25Radu Andrei Stefan nerelog25 Data 25 noiembrie 2017 11:52:24
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.82 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int t[100001],h[100001];
int localizare(int x)
{
  int r=x;
    while(t[r]) r=t[r];
    int y=x;
    while(y!=r) {int t1=t[y];t[y]=r;y=t1;}
    return r;
}
void unire(int x,int y)
{
    int c;
    if(h[x]<h[y]){t[x]=y;c=y;}
    else {t[y]=x;c=x;}
    if(h[x]==h[y]) h[c]++;
}
int main()
{
    int n,m,i,x,y,q;
    f>>n>>m;
    for(i=1; i<=n; i++)
    {
        t[i]=0;
        h[i]=1;
    }
    for(i=1; i<=m; i++)
    {
        f>>q>>x>>y;
        if(q==2)
        {
            if(localizare(x)==localizare(y)) g<<"DA"<<'\n';
            else g<<"NU"<<'\n';
        }
        else
        {

            unire(localizare(x),localizare(y));
        }
    }
    return 0;
}