Cod sursa(job #2083249)

Utilizator GeorgeCalinPetruta George-Calin GeorgeCalin Data 7 decembrie 2017 13:50:30
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 2.2 kb
#include <stdio.h>
#define si short int
using namespace std;
FILE *fin;
FILE *fout;

si t[6002];
si rg[6002];
int in_loc,out_loc;
char in_buff[4099],out_buff[500001];

inline void read_init(const char *nume)
{
    fin=fopen(nume,"r");
    in_loc=0;
}

inline void write_init(const char *nume)
{
    fout=fopen(nume,"w");
    out_loc=0;
}

inline void write_ch(char ch)
{
    if(out_loc==50000)
    {
        fwrite(out_buff,1,50000,fout);
        out_loc=0;
        out_buff[out_loc++]=ch;
    }
    else
        out_buff[out_loc++]=ch;
}

inline void write_int(int nr)
{
    if(nr<=9)
        write_ch(nr+'0');
    else
    {
        write_int(nr/10);
        write_ch(nr%10+'0');
    }
}

inline char read_ch()
{
    in_loc++;
    if(in_loc==4096)
    {
        in_loc=0;
        fread(in_buff,1,4096,fin);
    }
    return in_buff[in_loc];
}

inline int read_int()
{
    int nr=0;
    char c;
    while(!isdigit(c=read_ch())&&c!='-');
    int sgn=1;
    if(c==' ')
        sgn=-1;
    else
        nr=c-'0';
    while(isdigit(c=read_ch()))
        nr=nr*10+c-'0';
    return nr;
}

inline void write_appendix()
{
    fwrite(out_buff,1,out_loc,fout);
}

inline si radacina(si x)
{
    si r,y;
    r=x;
    while(r!=t[r])
    {
        r=t[r];
    }
    while(x!=t[x])
    {
        y=t[x];
        t[x]=r;
        x=y;
    }
    return r;
}

inline void unire(si x,si y)
{
    if(rg[x]>rg[y])
    {
        t[y]=t[x];
    }
    else
    {
        t[x]=t[y];
    }
    if(rg[x]==rg[y])
        rg[y]++;
}

int main()
{
    si n,x,c,y;
    read_init("entries.in");
    write_init("entries.out");
    n=read_int()
    for(int i=1;i<=5000;i++)
    {
        t[i]=i;
        rg[i]=1;
    }
    for(int i=1;i<=n;i++)
    {
        x=read_int();
        y=read_int();
        c=read_int();
        si radx=radacina(x);
        si rady=radacina(y);
        if(c==1)
        {
            unire(radx,rady);
        }
        else
        {
            c--;
            if(radx!=rady)
                c--;
            write_int(c);
            write_ch('\n');
        }
    }
    write_appendix();
    return 0;
}