Cod sursa(job #1087285)

Utilizator koroalinAlin Corodescu koroalin Data 19 ianuarie 2014 10:44:48
Problema Algoritmul lui Dijkstra Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.57 kb
#include <fstream>
#include <vector>
#define NMAX 50001
#define INFINIT 1000000000
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
vector <pair <int,int> > G[NMAX];
int n,prec[NMAX],m[NMAX],xs;
int cmin[NMAX];
void citire();
void cost();
void afisare();
void drumuri(int x);
int main()
{
    int i;
    citire();
    for (i=1; i<=n; i++) cmin[i]=INFINIT;
    for (i=0; i<G[xs].size(); i++) cmin[G[xs][i].first]=G[xs][i].second;
    for (i=1; i<=n; i++) prec[i]=xs;
    prec[xs]=0;
    m[xs]=1;
    cmin[xs]=0;
    cost();
    afisare();
    //for (i=1; i<=n; i++) {drumuri(i); fout<<'\n';}
    return 0;
}
void citire()
{ int i,x,y,m,c;
    fin>>n>>m;
    xs=1;
    for (i=0; i<m; i++)
    {
        fin>>x>>y>>c;
        G[x].push_back(make_pair(y,c));
    }
}
void cost()
{
    int i,minim,j,pozmin,x;
    for (i=1; i<n; i++)
    {
        minim=INFINIT+1;
        for (j=1; j<=n; j++) if (!m[j] && cmin[j]<minim) {minim=cmin[j]; pozmin=j;}
        m[pozmin]=1;
        x=pozmin;
        for (j=0; j<G[x].size(); j++)
        {
            if (!m[G[x][j].first] && cmin[x] + G[x][j].second< cmin[G[x][j].first])
            {
                cmin[G[x][j].first]=cmin[x] + G[x][j].second;
                prec[G[x][j].first]=x;
            }
        }
    }
}
void afisare()
{
    int i;
    for (i=2; i<=n; i++)
    {
        if (cmin[i]!=INFINIT) fout<<cmin[i]<<' ';
        else fout<<0<<' ';
    }

    fout<<'\n';
}
void drumuri(int x)
{
    if (x)

    {drumuri (prec[x]);
    fout<<x<<' ';}
}