Cod sursa(job #1647025)

Utilizator sicsicFMI-Coteanu Vlad sicsic Data 10 martie 2016 18:43:48
Problema Arbore partial de cost minim Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.9 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],T[Nmax],S=0;
bool U[Nmax],E[Nmax];
struct lista{int nod,cost; lista *leg; } *G[Nmax];
struct cmp
{
    bool operator()(const int &a, const int &b)
    {
        return D[a]>D[b];
    }
};
priority_queue<int,vector<int>,cmp> Q;
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,U[i]=0;
    D[1]=0; Q.push(1); E[1]=1;
    while(!Q.empty())
    {
        int nod=Q.top();
        Q.pop();
        E[nod]=0;
        S+=D[nod];
        U[nod]=1;
      //  g<<nod<<" "<<D[nod]<<'\n';
        for(lista *p=G[nod];p;p=p->leg)
        {
            if(p->cost<D[p->nod]&&!U[p->nod])
            {
                D[p->nod]=p->cost;
                T[p->nod]=nod;
                if(!E[p->nod])
                {
                   // g<<p->nod<<"*"<<'\n';
                    E[p->nod]=1,Q.push(p->nod);
                }
                else
                {
                    D[N+1]=-10002;
                    Q.push(N+1);
                    Q.pop();
                }
            }
        }
     /*   for(int i=1;i<=N;++i) g<<D[i]<<" ";
        g<<'\n';
        for(int i=1;i<=N;++i) g<<E[i]<<" ";
        g<<'\n';
        for(int i=1;i<=N;++i) g<<U[i]<<" ";
        g<<'\n';
        g<<'\n';*/
    }
}
void afis()
{
    //for(int i=2;i<=N;++i) S+=D[i];
   // g<<'\n';
    g<<S<<'\n';
    g<<N-1<<'\n';
    for(int i=1;i<=N;++i)
        if(T[i]) g<<i<<" "<<T[i]<<'\n';
}
int main()
{
    citire();
    APM();
    afis();
    return 0;
}