Cod sursa(job #2120644)

Utilizator malina2109Malina Diaconescu malina2109 Data 2 februarie 2018 18:51:57
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.94 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m;
int v[100010],rang[100010];
int cauta_rad(int x)
{
    int r=x,y;
    while(v[r]!=r) r=v[r];
    while(v[x]!=x)
    {
        y=v[x];
        v[x]=r;
        x=y;
    }
    return r;
}
void unire(int x, int y)
{
    if(rang[x]>rang[y])
        v[y]=x;
    else v[x]=y;
    if(rang[x]==rang[y]) rang[y]++;
}
int main()
{
    f>>n>>m;
    for(int i=1;i<=n;i++)
    {
        v[i]=i;
        rang[i]=1;
    }
    for(int i=1;i<=m;i++)
    {
        int op,x1,x2;
        f>>op>>x1>>x2;
        if(op==1)
        {
            int a=cauta_rad(x1);
            int b=cauta_rad(x2);
            if(a!=b)
                unire(a,b);
        }
        else
        {
            if(cauta_rad(x1)==cauta_rad(x2)) g<<"DA"<<"\n";
            else g<<"NU"<<"\n";
        }
    }
    return 0;
}