Cod sursa(job #1453181)

Utilizator al.mocanuAlexandru Mocanu al.mocanu Data 22 iunie 2015 21:23:49
Problema Algoritmul Bellman-Ford Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.14 kb
#include <stdio.h>
#include <queue>
#include <vector>
#define MAX 50005
#define INF 1<<20
using namespace std;

typedef struct{
	int node, cost;
} TNod;

int n, m, i, j, x, y, z, d[MAX], t[MAX];
bool viz[MAX];
vector<TNod> L[MAX];
queue<int> q;

bool bf(int s);

int main(){
	freopen("bellmanford.in", "r", stdin);
	freopen("bellmanford.out", "w", stdout);
	TNod aux;
	scanf("%d%d", &n, &m);
	for(i = 0; i < m; i++){
		scanf("%d%d%d", &x, &y, &z);
		aux.node = y;
		aux.cost = z;
		L[x].push_back(aux);
	}
	
	if(bf(1) == false)
		printf("Ciclu negativ!\n");
	else{
		for(i = 2; i <= n; i++)
			printf("%d ", d[i] == INF ? 0 : d[i]);
		printf("\n");
	}
	return 0;
}

bool bf(int s){
	vector<TNod>::iterator it;
	for(i = 1; i <= n; i++)
		d[i] = INF;
	d[s] = 0;
	q.push(s);
	viz[j] = true;

	while(!q.empty()){
		j = q.front();
		q.pop();
		viz[j] = false;
			
		it = L[j].begin();
		while(it < L[j].end()){
			if(d[it->node] > d[j] + it->cost){
				if(t[it->node] >= n - 1)
					return false;
				d[it->node] = d[j] + it->cost;
				if(viz[it->node] == false){
					viz[it->node] = true;
					q.push(it->node);
					t[it->node]++;
				}
			}
			it++;
		}
	}
	return true;
}