Cod sursa(job #1360739)

Utilizator ooptNemes Alin oopt Data 25 februarie 2015 17:33:39
Problema Algoritmul Bellman-Ford Scor 35
Compilator cpp Status done
Runda Arhiva educationala Marime 1.13 kb
// bellmanford
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>

#define pb push_back
#define inf (1<<30)+100
#define NMax 50005

using namespace std;

ifstream f("bellmanford.in");
ofstream g("bellmanford.out");

int n, m;
vector<int> V[NMax];
vector<int> C[NMax];
int dist[NMax];
queue<int> q;

void read() {
	f>>n>>m;
	for (int i=1;i<=m;i++) {
		int a, b, c;
		f>>a>>b>>c;
		V[a].pb(b);
		C[a].pb(c);
	}
}

void solve() {
	for (int i=1;i<=n;i++)
		dist[i] = inf;
	dist[1] = 0;

	int added = 1;
	q.push(1);
	while (!q.empty() && added <= 1LL*n*m) {
		int nod = q.front();
		q.pop();

		for (unsigned i=0;i<V[nod].size();i++) {
			int vecin = V[nod][i];
			if (dist[vecin] > dist[nod] + C[nod][i]) {
				dist[vecin] = dist[nod] + C[nod][i];
				q.push(vecin);
				added++;
			}
		}
	}

	for (int i=1;i<=n;i++) {
		for (unsigned j=0;j<V[i].size();j++) {
			int nod = i;
			int vecin = V[i][j];

			if (dist[vecin] > dist[nod] + C[i][j]) {
				cout<<"Ciclu negativ!\n";
				return;
			}
		}
	}

	for (int i=2;i<=n;i++)
		g<<((dist[i]==inf)?0:dist[i])<<' ';
}

int main() {

	read();
	solve();

	f.close(); g.close();
	return 0;
}