Pagini recente » Cod sursa (job #898426) | Cod sursa (job #3289792) | Cod sursa (job #1830606) | Cod sursa (job #3036369) | Cod sursa (job #2481239)
#include<fstream>
#include<set>
#include<vector>
#include<iostream>
#define DIM 5010
#define INF DIM*1000
using namespace std;
vector<pair<int, int>> adjacency_list[DIM];
set<pair<int, int>> tail;
int distances[DIM], no_vertices, no_edges, i, x, y, c;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
int main()
{
fin >> no_vertices >> no_edges;
for (i = 1; i <= no_vertices; i++)
{
fin >> x >> y >> c;
adjacency_list[x].push_back(make_pair(y, c));
}
for (i = 1; i <= no_vertices; i++)
distances[i] = INF;
distances[1] = 0;
tail.insert(make_pair(0, 1));
while (!tail.empty())
{
int vertex = tail.begin()->second;
tail.erase(tail.begin());
for (i = 0; i < adjacency_list[vertex].size(); i++)
{
int neighbour = adjacency_list[vertex][i].first;
int cost = adjacency_list[vertex][i].second;
if (distances[neighbour] > distances[vertex] + cost)
{
tail.erase(make_pair(distances[neighbour], neighbour));
distances[neighbour] = distances[vertex] + cost;
tail.insert(make_pair(distances[neighbour], neighbour));
}
}
}
for (i = 2; i <= no_vertices; i++)
{
if (distances[i] == INF)
fout << 0 << " ";
else
fout << distances[i] << " ";
}
return 0;
}