Pagini recente » Cod sursa (job #1021733) | Cod sursa (job #1981617) | Cod sursa (job #1025456) | Cod sursa (job #964466) | Cod sursa (job #2076933)
#include <iostream>
#include <fstream>
#include <vector>
#include <utility>
#include <set>
#define INFINITY 2147483647
#define MAX_N 50001
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
vector<pair<int, int>> graf[MAX_N];
int dist[MAX_N];
int n, m;
int source;
set<pair<int, int>> q;
int main() {
fin>>n>>m;
source = 1;
for (int i = 0; i < m; i++) {
int a, b, c;
fin>>a>>b>>c;
graf[a].push_back(make_pair(c, b));
}
for (int i = 1; i <= n; i++) {
dist[i] = INFINITY;
if (i == source)
dist[i] = 0;
q.insert(make_pair(dist[i], i));
}
while (!q.empty()) {
pair<int, int> nod = *q.begin();
q.erase(q.begin());
vector<pair<int, int>> neighbours = graf[nod.second];
vector<pair<int, int>>::iterator it;
for (it = neighbours.begin(); it != neighbours.end(); it++) {
pair<int, int> neighbour = *it;
if (dist[nod.second] != INFINITY) {
if (q.find(make_pair(dist[neighbour.second], neighbour.second)) != q.end()) {
int alt = nod.first + neighbour.first;
if (alt < dist[neighbour.second]) {
q.erase(make_pair(dist[neighbour.second], neighbour.second));
dist[neighbour.second] = alt;
q.insert(make_pair(dist[neighbour.second], neighbour.second));
}
}
}
}
}
for (int i = 1; i <= n; i++) {
if (i != source) {
if (dist[i] == INFINITY)
fout<<"0 ";
else
fout<<dist[i]<<" ";
}
}
return 0;
}