Cod sursa(job #1966248)

Utilizator GeoeyMexicanuBadita George GeoeyMexicanu Data 15 aprilie 2017 01:11:35
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.98 kb
#include <iostream>
#include <fstream>
#define N 100020

using namespace std;

ifstream f("disjoint.in");
ofstream g("disjoint.out");

int T[N],Rg[N];
int i,j,n,m,k,t,p,r;
int gasesc(int x)
{
    int R,y;
    for(R=x;T[R]!=R;R=T[R]);
    for(;T[x]!=x;)
    {
        y=T[x];
        T[x]=R;
        x=y;
    }
    return R;
}
void unesc(int x,int y)
{
    if(Rg[x]>Rg[y])
        T[y]=x;
    else
        T[x]=y;
    if(Rg[x]==Rg[y])
        Rg[y]++;
}
int main()
{
    int x,y,cd;
    f>>n>>m;
    for(int i=1;i<=n;i++)
    {
        T[i]=i;
        Rg[i]=1;
    }
    for(i=1;i<=m;i++)
    {
        f>>cd>>x>>y;
        if(cd==2)
        {
            if(gasesc(x)==gasesc(y))
                g<<"DA"<<"\n";
            else
                g<<"NU"<<"\n";
        }
        else
        {
            if(gasesc(x)==gasesc(y))
            {
                return 0;
            }
            unesc(gasesc(x),gasesc(y));
        }
    }
}