Cod sursa(job #781013)

Utilizator oana_popfmi - pop oana oana_pop Data 22 august 2012 23:23:53
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.17 kb
#include <fstream>
#include <iostream>
#include <set>
#include <vector>
#define MAXN 50100
#define INF 1000000000
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");

int n , m , d[MAXN];
vector<int> g[MAXN],c[MAXN];
set < pair <int,int > > t;
 
void dijkstra()
{
     int i , val , x;
     for(i=2 ; i<=n; i++) d[i]=INF;

     t.insert(make_pair(0,1));
     while(t.size()>0)
     {
                      val=(*t.begin()).first; 
                      x=(*t.begin()).second; 
                      t.erase(*t.begin());
                      for(i=0;i<g[x].size(); i++)
                       if(d[g[x][i]] > val + c[x][i])
                         {
                                      d[g[x][i]]=val+c[x][i];
                                      t.insert(make_pair(d[g[x][i]],g[x][i]));
                         }
     }
} 

int main()
{
    int i , a ,b , cost;
    fin>>n>>m;
    for(i=1; i<=m; i++) 
    {
             fin>>a>>b>>cost;
             g[a].push_back(b);
             c[a].push_back(cost);
    }
    dijkstra();
    for(i=2; i<=n; i++)
    if(d[i]==INF) fout<<"0"<<" ";
    else fout<<d[i]<<" ";


}