Cod sursa(job #2296142)

Utilizator lucianistratiIstrati Lucian lucianistrati Data 4 decembrie 2018 14:58:05
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.58 kb
#include <iostream>
#include <fstream>
using namespace std;
int v[100001],rang[100001];
int cautare(int x)
{
    int a,aux;
    a=x;
    while(v[a]!=a)//cat timp v[a]!=a
    {
        a=v[a];//atribuim lui a aceasta valoarea
    }
    if(v[x]!=x)
    {
        aux=v[x];//realizam o interschimbare intre v[x] si v[a]
        v[x]=a;//atribuim lui v[x] a care este egal cu v[a]
        x=aux;
    }
    return a;
}
void reuniune(int x,int y)
{
    if(rang[x]>rang[y])//daca valoarea lui rang[x]>rang[y]
    {
        v[y]=x;//introducem in v[y] valoarea lui x
    }
    else
    {
        v[x]=y;//altfel in v[x] valoarea lui y
    }
    if(rang[x]==rang[y])//daca cele doua rang[x] si rang[y] sunt egale
    {
        rang[y]++;//incrementam rang[y]
    }
}
int main()
{
    ifstream fin("disjoint.in");
    ofstream fout("disjoint.out");
    int i,j,N,M,cod,x,y;
    fin>>N>>M;
    for(i=1;i<=N;i++)
    { 
        v[i]=i; //initializam elementele din vectorul v cu i
        rang[i]=1;//initializam vectorul rang cu fiecare element=1
    }
    for(i=1;i<=M;i++)
    {
    fin>>cod>>x>>y;
    if(cod==1)
    {
        reuniune(cautare(x),cautare(y));//reuneste cele doua elemente
        //in aceeasi multime
    }
    if(cod==2)
    {
        if(cautare(x)==cautare(y))//daca cele doua elemente apartin aceleiasi multime
            //atunci afisa "DA", altfel afisam "NU"
        {
            fout<<"DA\n";
        }
        else
        {
            fout<<"NU\n";
        }
    }
    }
    fin.close();
    fout.close();
    return 0;
}