Cod sursa(job #1484728)

Utilizator howsiweiHow Si Wei howsiwei Data 11 septembrie 2015 18:58:32
Problema Algoritmul Bellman-Ford Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.43 kb
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <queue>
#include <algorithm>
#include <functional>
using namespace std;

const int oo = 1<<29;

int main() {
	ios::sync_with_stdio(false);
	freopen("bellmanford.in", "r", stdin);
	freopen("bellmanford.out", "w", stdout);
	int n, m;
	cin >> n >> m;
	vector<vector<pair<int,int>>> adjl(n);
	for (int i = 0; i < m; i++) {
		int u, v, w;
		cin >> u >> v >> w;
		u--, v--;
		adjl[u].emplace_back(w, v);
	}

	vector<int> dist(n, oo);
	dist[0] = 0;
	queue<int> q;
	q.push(0);
	vector<bool> inq(n);
	vector<int> par(n);
	vector<vector<int>> subss(n);
	vector<bool> ancinq(n);

	function<bool(int,int)> issub = [&](int w, int u) {
		ancinq[w] = true;
		bool ret = w == u;
		remove_if(subss[w].begin(), subss[w].end(), [&](int x) { return par[x] != w; });
		for (auto x: subss[w]) {
			ret = ret || issub(x, u);
		}
		return ret;
	};

	do {
		auto u = q.front();
		q.pop();
		inq[u] = false;
		if (ancinq[u]) {
			continue;
		}
		for (auto e: adjl[u]) {
			auto v = e.second;
			auto w = e.first;
			if (dist[v] > dist[u]+w) {
				dist[v] = dist[u]+w;
				subss[u].push_back(v);
				par[v] = u;
				if (!inq[v]) {
					if (issub(v, u)) {
						puts("Ciclu negativ!");
						return 0;
					}
					q.push(v);
					inq[v] = true;
					ancinq[v] = false;
				}
			}
		}
	} while (!q.empty());

	for (int i = 1; i < n; i++) {
		printf("%d%s", dist[i], i < n-1 ? " " : "\n");
	}
	return 0;
}