Cod sursa(job #401799)

Utilizator arnold23Arnold Tempfli arnold23 Data 23 februarie 2010 09:28:27
Problema Algoritmul lui Dijkstra Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.08 kb
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

#define INF 0x3f3f3f3f
const long size = 50100;
long n,m,i,u,x,y,t;
long dist[size];

struct nod{
 long csp,ta;
};

vector<nod> v[size];
vector<nod>::iterator q;
nod adat;

class comp{
public:
    bool operator()(long& a, long& b)
    {
      return dist[a]<dist[b];
    }
};

priority_queue<long, vector<long>, comp> heap;

int main()
{
 ifstream in("dijkstra.in");
 ofstream out("dijkstra.out"); 
 
 in >> n >> m;
 for(i=1;i<=m;++i){
   in >> x >> y >> t;
   adat.csp=y;
   adat.ta=t;
   v[x].push_back(adat);                
 }                 
 
 memset(dist,INF,sizeof(dist));
 dist[1]=0;
 heap.push(1);

 while(!heap.empty()){
  u=heap.top();
  heap.pop();
  for(q=v[u].begin(); q<v[u].end(); ++q){
   if(dist[q->csp]>dist[u]+q->ta){
      dist[q->csp]=dist[u]+q->ta;
      heap.push(q->csp);                              
   }             
  } 
 }   
 
 for(i=2;i<=n;++i){
    if(dist[i]==INF) out << "0 ";  
    else out << dist[i] << " "; 
 }   
 
 return 0;
}