Cod sursa(job #1365473)

Utilizator firutibogdanFiruti Bogdan-Cristian firutibogdan Data 28 februarie 2015 12:07:41
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.1 kb
#include<fstream>
using namespace std;
int n,m,a,b,c,i,vecin[100003],r1,r2,rang[100003];
int cautare(int x)
{
    int r,y;
    for(r=x;vecin[r]!=r;r=vecin[r]);

    while(vecin[x] != x)
    {
        y = vecin[x];
        vecin[x] = r;
        x = y;
    }
    return r;
}
void unesc(int x,int y)
{
    if(rang[x]>rang[y])
    {
        vecin[y]=x;
    }
    else
    {
        vecin[x]=y;
    }
    if(rang[x]==rang[y])
    {
        rang[y]++;
    }
}
int main()
{
     ifstream fin("disjoint.in");
    ofstream fout("disjoint.out");
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        vecin[i]=i;
        rang[i]=1;
    }
    for(i=1;i<=m;i++)
    {
        fin>>c>>a>>b;
        if(c==2)
        {
            r1=cautare(a);
            r2=cautare(b);
            if(r1==r2)
            {
                fout<<"DA"<<"\n";
            }
            else
            {
                fout<<"NU"<<"\n";
            }
        }
        else
        {
            unesc(cautare(a),cautare(b));
        }
    }
    fin.close();
    fout.close();
    return 0;
}