Cod sursa(job #1791895)

Utilizator Emil64Emil Centiu Emil64 Data 29 octombrie 2016 20:16:19
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.41 kb
#include <stdio.h>
#include <algorithm>

using namespace std;

int v[400001]={0};
int dad[200001]={0};

struct arcu{
    int a, b, cost;
}arc[400001]={0};

bool cmp(arcu a1, arcu a2){
    return (a1.cost<a2.cost);
}

int find_dad(int nod){

    if(dad[nod] != nod)
        dad[nod] = find_dad(dad[nod]);
    return dad[nod];
}

char buff[20000];int pos=0, semn;
FILE*f=freopen("apm.in","r",stdin);
FILE*g=freopen("apm.out","w",stdout);
inline void read(int &nr){
    semn = 1;
    while(buff[pos] < '0' || buff[pos] > '9'){if(buff[pos]== '-' )semn = -1; if(++pos == 20000) fread(buff, 1, 20000, stdin), pos = 0;}
    nr = 0;
    while('0' <= buff[pos] && buff[pos] <= '9') {nr = nr * 10 + buff[pos] - '0';if(++pos == 20000) fread(buff, 1, 20000, stdin), pos = 0;}
    nr*=semn;
}

int main()
{
    fread(buff, 1, 20000, stdin);
    int i, j, n, m, op, a, b, ta, tb, cost=0, lg=0;
    read(n);
    read(m);
    for(i=1;i<=m;i++){

        read(arc[i].a);read(arc[i].b);read(arc[i].cost);
    }

    for(i=1;i<=n;i++)
        dad[i]=i;

    sort(arc+1, arc+1+m, cmp);

    for(i=1;i<=m;i++)
    {
        ta=find_dad(arc[i].a);
        tb=find_dad(arc[i].b);
        if(ta!=tb)
        {
            cost+=arc[i].cost;
            v[++lg]=i;
            dad[ta]=tb;
            if(lg>=(n-1))
                break;
        }
    }
    printf("%d\n%d\n", cost, lg);
    for(i=1;i<n;i++){
        printf("%d %d\n", arc[v[i]].a, arc[v[i]].b);
    }
}