Cod sursa(job #2670180)

Utilizator retrogradLucian Bicsi retrograd Data 9 noiembrie 2020 12:13:00
Problema Algoritmul Bellman-Ford Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.89 kb
#include <bits/stdc++.h>

using namespace std;

class InParser {
private:
  FILE* fin;
  char* buff;
  int sp;

  char read_ch() {
    ++sp;
    if (sp == 4096) {
      sp = 0;
      fread(buff, 1, 4096, fin);
    }
    return buff[sp];
  }

public:
  InParser(const char* nume) {
    fin = fopen(nume, "r");
    buff = new char[4096]();
    sp = 4095;
  }

  InParser& operator >> (int& n) {
    char c;
    while (!isdigit(c = read_ch()) && c != '-');
    int sgn = 1;
    if (c == '-') {
      n = 0;
      sgn = -1;
    } else {
      n = c - '0';
    }
    while (isdigit(c = read_ch())) {
      n = 10 * n + c - '0';
    }
    n *= sgn;
    return *this;
  }
};

int main() {
  InParser cin("bellmanford.in");
  ofstream cout("bellmanford.out");

  int n, m; cin >> n >> m;
  vector<tuple<int, int, int>> es(m);
  for (int i = 0; i < m; ++i) {
    int a, b, c; cin >> a >> b >> c; --a; --b;
    es[i] = make_tuple(a, b, c);
  }
  sort(es.begin(), es.end());
  vector<int> start(n + 1, m);
  for (int i = m - 1; i >= 0; --i)
    start[get<0>(es[i])] = i;
  for (int i = n - 1; i >= 0; --i)
    start[i] = min(start[i], start[i + 1]);

  vector<int> dist(n, 1e9), prev(n, -1),
    q(n, -1), inq(n, false);
  int b = 0, e = 0;
  auto push = [&](int node, int d, int p) {
    if (dist[node] <= d) return;
    dist[node] = d;
    prev[node] = p;
    if (!inq[node]) {
      q[e] = node;
      if (++e == n)
        e = 0;
    }
    inq[node] = true;
  };

  push(0, 0, -1);
  vector<int> cnt(n, 0);

  while (b != e) {
    int node = q[b];
    if (cnt[node]++ == n) {
      cout << "Ciclu negativ!" << endl;
      return 0;
    }
    if (++b == n) b = 0;
    inq[node] = false;

    for (int i = start[node]; i < start[node + 1]; ++i) {
      int vec, cost; tie(ignore, vec, cost) = es[i];
      push(vec, cost + dist[node], node);
    }
  }

  for (int i = 1; i < n; ++i)
    cout << dist[i] << " ";
  cout << endl;

  return 0;
}