Cod sursa(job #1196704)

Utilizator bogdanmarin69Bogdan Marin bogdanmarin69 Data 8 iunie 2014 22:03:10
Problema Algoritmul Bellman-Ford Scor 35
Compilator cpp Status done
Runda Arhiva educationala Marime 1.07 kb
#include <cstdio>
#include <vector>
using namespace std;
#define MAX 50001
#define INF 1000000000
vector <pair<int, int> > lista[MAX];
bool ok[MAX];
int n, m, dist[MAX], tata[MAX], st[MAX], vf, relaxari[MAX];
int bellman(int s);
int main()
{
	int i, x, y, w;
	freopen("bellmanford.in", "r", stdin);
	freopen("bellmanford.out", "w", stdout);
	scanf("%d%d", &n, &m);
	for(i=1; i<=m; i++){
		scanf("%d%d%d", &x, &y, &w);
		lista[x].push_back(make_pair(y, w));
	}
	if(bellman(1)==0)
		printf("Ciclu negativ!\n");
	else
		for(i=2; i<=n; i++)
			printf("%d ", dist[i]);
	return 0;
}
int bellman(int s)
{
	int i, gata, iter=0, j, y, w;
	for(i=1; i<=n; i++){
		dist[i] = INF;
		tata[i] = 0;
	}
	dist[s] = 0;
	vf=1; st[vf] = 1;
	while(vf>0){
		i = st[vf--];
		relaxari[i]++;
		if(relaxari[i]==n) return 0;
		ok[i] = false;
		for(j=0; j<lista[i].size(); j++){
			y = lista[i][j].first;
			w = lista[i][j].second;
			if(dist[i]+w<dist[y]){
				dist[y] = dist[i]+w;
				if(ok[y]==false){
					ok[y] = true;
					st[++vf] = y;
				}
			}
		}
	}
	return 1;
}