Pagini recente » Cod sursa (job #2724657) | Cod sursa (job #2363642) | monthly-2012/runda-8/solutii | Cod sursa (job #499971) | Cod sursa (job #2611139)
// @ Copyright : Darius Neatu
#include <bits/stdc++.h>
#include <iostream>
#include <utility>
#define NMAX 50010
#define oo (1 << 30)
#define NO_PARENT (-1)
using namespace std;
class Task {
public:
void solve() {
read_input();
get_result();
print_output();
}
private:
int n, m;
int source;
vector<pair<int, int> > adj[NMAX];
priority_queue<pair<int, int>, vector<pair<int, int> >,
std::greater<pair<int, int> > > pq;
vector<int> d;
vector<int> p;
void read_input() {
cin >> n >> m;
d.resize(n + 1);
p.resize(n + 1);
source = 1;
for (int i = 1; i <= m; ++i) {
int x, y, c;
cin >> x >> y >> c;
adj[x].push_back({y, c});
}
}
void get_result() { Dijkstra(source, d, p); }
void Dijkstra(int source, vector<int> &d, vector<int> &p) {
for (int i = 1; i <= n; ++i) {
d[i] = oo;
p[i] = NO_PARENT;
}
p[source] = 0;
d[source] = 0;
pq.push({d[source], source});
while (!pq.empty()) {
auto entry = pq.top();
pq.pop();
int cost = entry.first;
int node = entry.second;
if (cost > d[node]) {
continue;
}
for (auto &edge : adj[node]) {
int neighbour = edge.first;
int cost_edge = edge.second;
if (d[node] + cost_edge < d[neighbour]) {
d[neighbour] = d[node] + cost_edge;
p[neighbour] = node;
pq.push({d[neighbour], neighbour});
}
}
}
for (int i = 1; i <= n; ++i) {
if (d[i] == oo) {
d[i] = 0;
}
}
}
void print_output() {
for (int i = 1; i <= n; ++i) {
if (i != source) {
cout << (d[i] == oo ? 0 : d[i])<<" ";
}
//auto path = rebuild_path(i, p);
}
}
};
int main() {
auto cin_buff = cin.rdbuf();
auto cout_buff = cout.rdbuf();
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
cin.rdbuf(fin.rdbuf());
cout.rdbuf(fout.rdbuf());
Task *task = new Task();
task->solve();
delete task;
cin.rdbuf(cin_buff);
cout.rdbuf(cout_buff);
return 0;
}