Cod sursa(job #1643306)

Utilizator sicsicFMI-Coteanu Vlad sicsic Data 9 martie 2016 18:26:32
Problema Arbore partial de cost minim Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.19 kb
#include<fstream>
#include<queue>
using namespace std;
#define Nmax 200003
#define INF 9999999
ifstream f("apm.in");
ofstream g("apm.out");
int N,M,D[Nmax],S2=0,T[Nmax];
bool U[Nmax];
struct lista{int nod,cost; lista *leg; } *G[Nmax];
void adaug(int i,int j,int c)
{
    lista *p;
    p=new lista;
    p->nod=j;
    p->leg=G[i];
    p->cost=c;
    G[i]=p;
}
void citire()
{
    f>>N>>M;
    int i,j,c;
    while(M--)
    {
        f>>i>>j>>c;
        adaug(i,j,c);
        adaug(j,i,c);
    }
}
void APM()
{
    for(int i=1;i<=N;++i) D[i]=INF;
    for(lista *p=G[1];p;p=p->leg) D[p->nod]=p->cost,T[p->nod]=1;
    U[1]=1;
    while(1)
    {
        int mini=INF,poz;
        for(int i=1;i<=N;++i) if(D[i]<mini&&!U[i]) mini=D[i],poz=i;
        if(mini==INF) break;
        U[poz]=1;
        S2+=D[poz];
        for(lista *p=G[poz];p;p=p->leg)
            if(D[p->nod]>p->cost&&!U[p->nod])
            {
                D[p->nod]=p->cost;
                T[p->nod]=poz;
            }

    }
}
void afis()
{
    g<<S2<<'\n';
    g<<N-1<<'\n';
    for(int i=2;i<=N;++i) g<<T[i]<<" "<<i<<'\n';
}
int main()
{
    citire();
    APM();
    afis();
    return 0;
}