Cod sursa(job #1077065)

Utilizator pulseOvidiu Giorgi pulse Data 10 ianuarie 2014 20:50:41
Problema Algoritmul Bellman-Ford Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.04 kb
#include <fstream>
#include <vector>
#include <queue>
#define INF 0x3f3f3f3f
#define NMAX 50002

using namespace std;

ifstream fin ("bellmanford.in"); ofstream fout ("bellmanford.out");

int N, M, cost[NMAX], nr[NMAX], ok;
vector <int> v[NMAX];
queue <int> q;

void ReadData ()
{
	fin >> N >> M;
	for (int i = 1, a, b, c; i <= M; ++i)
	{
		fin >> a >> b >> c;
		v[a].push_back (b);
		v[a].push_back (c);
	}
	for (int i = 2; i <= N; ++i)
		cost[i] = INF;
}

int Solve ()
{
	q.push (1);
	while (!q.empty ())
	{
		int x = q.front ();
		for (int i = 0; i < v[x].size (); i += 2)
		{
			if (cost[v[x][i]] > cost[x] + v[x][i + 1])
			{
				q.push (v[x][i]);
				cost[v[x][i]] = cost[x] + v[x][i + 1];
				++nr[v[x][i]];
			}
			if (nr[v[x][i]] >= N)
				return 0;
		}
		q.pop ();
	}
	return 1;
}

void WriteData ()
{
	if (!ok)
		fout << "Ciclu negativ!\n";
	else
	{
		for (int i = 2; i <= N; ++i)
			fout << cost[i] << " ";
	}
	fout << "\n";
}

int main ()
{
	ReadData ();
	ok = Solve ();
	WriteData ();
}