Pagini recente » Cod sursa (job #2396613) | Cod sursa (job #2947817) | Cod sursa (job #1986382) | Cod sursa (job #2848777) | Cod sursa (job #2611137)
#include <bits/stdc++.h>
const int kNmax = 50009;
struct Edge {
int x;
int y;
};
class Task {
public:
void solve() {
read_input();
get_result();
}
private:
int n;
int m;
std::vector<std::pair<int, int>> adj[kNmax];
std::vector<int> distances;
std::priority_queue <std::pair<int, int>, std::vector<std::pair<int, int>>,
std::greater<std::pair<int, int>>> pq;
void read_input() {
std::ifstream fin("dijkstra.in");
fin >> n >> m;
for (int i = 1, x, y, cost; i <= m; i++) {
fin >> x >> y >> cost;
adj[x].push_back(std::pair<int, int>(cost, y));
}
fin.close();
}
void get_result() {
distances = std::vector<int>(n + 1, INT_MAX);
std::pair<int, int> elem(0, 1);
pq.push(elem);
distances[1] = 0;
while (!pq.empty()) {
auto aux = pq.top();
int node = aux.second;
int c = aux.first;
pq.pop();
if (c > distances[node]) {
continue;
}
for (auto& it : adj[node]) {
int adj = it.second;
int cost = it.first;
if (distances[adj] > distances[node] + cost) {
distances[adj] = distances[node] + cost;
std::pair<int, int> newElem(distances[adj], adj);
pq.push(newElem);
}
}
}
std::ofstream fout("dijkstra.out");
for (int i = 2; i <= n; i++) {
if (distances[i] == INT_MAX) {
fout << "0 ";
} else {
fout << distances[i] << " ";
}
}
fout << "\n";
fout.close();
}
};
int main() {
Task *task = new Task();
task->solve();
delete task;
return 0;
}