Pagini recente » Cod sursa (job #2073683) | Cod sursa (job #165840) | Solutii Autumn Warmup, Runda 3 | Cod sursa (job #1750483) | Cod sursa (job #2982891)
#include <fstream>
#include <vector>
#include <set>
#define DIM 50005
#define INF 0x3f3f3f3f
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
struct elem {
int node, cost;
bool operator < (elem x) const {
return cost < x.cost;
}
};
int n, m, dist[DIM];
vector<pair<int, int>> edges[DIM];
set<pair<int, int>> s;
void dijkstra(int node) {
dist[1] = 0;
s.insert({0, node});
while (!s.empty()) {
int node = s.begin() -> second;
s.erase(s.begin());
for (auto child: edges[node]) {
if (dist[child.first] > dist[node] + child.second) {
s.erase({dist[child.first], child.first});
dist[child.first] = dist[node] + child.second;
s.insert({dist[child.first], child.first});
}
}
}
}
int main() {
f >> n >> m;
for (int i = 1; i <= m; i++) {
int x, y, c;
f >> x >> y >> c;
edges[x].push_back({y, c});
}
for (int i = 1; i <= n; i++) {
dist[i] = INF;
}
dijkstra(1);
for (int i = 2; i <= n; i++) {
if (dist[i] == INF) {
g << 0 << " ";
} else {
g << dist[i] << " ";
}
}
return 0;
}