Cod sursa(job #2296019)

Utilizator lucianistratiIstrati Lucian lucianistrati Data 4 decembrie 2018 09:29:47
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.02 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)
    {
        a=v[a];
    }
    if(v[x]!=x)
    {
        aux=v[x];
        v[x]=a;
        x=aux;
    }
    return a;
}
void reuniune(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()
{
    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;
        rang[i]=1;
    }
    for(i=1;i<=M;i++)
    {
    fin>>cod>>x>>y;
    if(cod==1)
    {
        reuniune(cautare(x),cautare(y));
    }
    if(cod==2)
    {
        if(cautare(x)==cautare(y))
        {
            fout<<"DA\n";
        }
        else
        {
            fout<<"NU\n";
        }
    }
    }
    fin.close();
    fout.close();
    return 0;
}