Cod sursa(job #1525088)

Utilizator sicsicFMI-Coteanu Vlad sicsic Data 14 noiembrie 2015 18:50:09
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.77 kb
#include<fstream>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m,T[100007],C[100007],op,x,y;
int rad(int i)
{
    int k=i;
    while(T[k]!=k) k=T[k];
    while(T[i]!=i)
    {
        int j=T[i];
        T[i]=k;
        i=j;
    }
    return k;
}
void uneste(int i,int j)
{
    if(C[i]>=C[j])
    {
        C[i]+=C[j];
        T[j]=i;
        C[j]=0;
    }
    else
    {
        C[j]+=C[i];
        C[i]=0;
        T[i]=j;
    }
}
int main()
{
    f>>n>>m;
    for(int i=1;i<=n;++i) T[i]=i,C[i]=1;
    for(int i=1;i<=m;++i)
    {
        f>>op>>x>>y;
        int rx=rad(x),ry=rad(y);
        if(op==1) uneste(rx,ry);
        else if(rx==ry) g<<"DA"<<'\n';
        else g<<"NU"<<'\n';
    }
    return 0;
}