Pagini recente » Cod sursa (job #623315) | Cod sursa (job #2883198) | Cod sursa (job #3271118) | Cod sursa (job #742432) | Cod sursa (job #2887753)
#include <iostream>
#include <fstream>
#include <vector>
#include <set>
using namespace std;
ifstream fin("bellmanford.in");
ofstream fout("bellmanford.out");
struct Edge {
int b, v;
};
struct Node {
int a, v;
};
bool operator<(const Node &lhs, const Node &rhs) {
return lhs.v < rhs.v || (lhs.v == rhs.v && lhs.a < rhs.a);
}
const int INF = 0x3f3f3f3f;
const int N = 50041;
const int M = 250041;
int n, m;
vector<Edge> gra[N];
int d[N], c[N];
int main() {
fin >> n >> m;
for (int i = 0; i < m; ++i) {
int a, b, v;
fin >> a >> b >> v;
gra[a].push_back({b, v});
}
for (int i = 1; i <= n; ++i) {
d[i] = INF;
}
bool negative_cycle = false;
set<Node> se;
d[1] = 0;
se.insert({1, 0});
while (!negative_cycle && !se.empty()) {
auto node = *se.begin(); se.erase(se.begin());
for (auto edge : gra[node.a]) {
if (node.v + edge.v < d[edge.b]) {
if (d[edge.b] == INF) { se.erase({edge.b, d[edge.b]}); }
d[edge.b] = node.v + edge.v;
se.insert({edge.b, node.v + edge.v});
c[edge.b]++;
if (c[edge.b] >= n) { negative_cycle = true; }
}
}
}
if (negative_cycle) {
fout << "Ciclu negativ!";
} else {
for (int i = 2; i <= n; ++i) {
fout << d[i] << " ";
}
}
return 0;
}