Pagini recente » Cod sursa (job #1090604) | Cod sursa (job #417870) | exemplu12 | Cod sursa (job #557974) | Cod sursa (job #2401123)
#include <iostream>
#include <queue>
#include <vector>
#include <fstream>
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
#define inf 100000000
#define NMAX 50003
vector< int > graph[NMAX], graphC[NMAX];
priority_queue <pair <int, int> > myheap;
int n, m, dist[NMAX], viz[NMAX];
void citire()
{
f>>n>>m;
for(int i = 0; i < m; i ++)
{
int a, b, c;
f>>a>>b>>c;
/// muchie de la a->b
graph[a].push_back(b);
/// costul muchiei de la a->b
graphC[a].push_back(c);
}
}
void algoritm()
{
int index;
for(int i = 1; i <= n; i ++)
dist[i] = inf;
int sursa = 1;
dist[sursa] = 0;
myheap.push(make_pair(0, sursa));
for(int i = 2; i <= n; i ++)
myheap.push(make_pair(-inf, i));
while(!myheap.empty())
{
pair<int, int> best = myheap.top();
myheap.pop();
index = best.second;
if(viz[index] == 0){
int nrVecini = graph[index].size();
viz[index] = 1;
/// parcurg vecinii nodului index;
for(int j = 0; j < nrVecini; j ++)
{
int vecin = graph[index][j];
int cost = graphC[index][j];
if((dist[index] + cost) < dist[vecin])
{
dist[vecin] = dist[index] + cost;
myheap.push(make_pair(-dist[vecin], vecin));
}
}
}
}
}
void afisare()
{
for(int i = 2; i <= n; i ++)
if(dist[i] == inf)
g<<0<<" ";
else g<<dist[i]<<" ";
}
int main()
{
citire();
algoritm();
afisare();
return 0;
}