Cod sursa(job #1247205)

Utilizator DanyPrvPirvoaica Daniel DanyPrv Data 22 octombrie 2014 12:59:52
Problema Algoritmul Bellman-Ford Scor 5
Compilator cpp Status done
Runda Arhiva educationala Marime 1.05 kb
#include <fstream>
#include <vector>
#include <limits.h>
using namespace std;
ifstream f("bellmanford.in");
ofstream g("bellmanford.out");
vector <pair<int, int> >a[50001];
int m,n,x,y,z,p,u,q[50001],nre[50001],c[50001],viz[50001],i,nrap[50001];
int main()
{
	f>>n>>m;
	for(int i=1;i<=m;i++){
		f>>x>>y>>z;
		a[x].push_back(make_pair(y,z));
		nre[x]++;
	}
	for(i=2;i<=n;i++)
		c[i]=INT_MAX;
	p=u=q[1]=viz[1]=1;

	while(p<=u){
		for(i=0;i<nre[q[p]];i++){
			if(!viz[a[q[p]][i].first]&&nrap[a[q[p]][i].first]==0){
				nrap[q[p]]++;
				viz[a[q[p]][i].first]=1;	
				q[++u]=a[q[p]][i].first;
				if(a[q[p]][i].second+c[q[p]]<c[a[q[p]][i].first])
					c[a[q[p]][i].first]=a[q[p]][i].second+c[q[p]];
			}
			else{
				nrap[a[q[p]][i].first]++;
				if(nrap[a[q[p]][i].first]<n){
					if(a[q[p]][i].second+c[q[p]]<c[a[q[p]][i].first])
					c[a[q[p]][i].first]=a[q[p]][i].second+c[q[p]];
				}
				else{
					g<<"Ciclu negativ!";
					return 0;
				}
			}
		}
		viz[q[p]]=0;
		p++;
	}
	for(i=2;i<=n;i++)
		g<<c[i]<<' ';
    return 0;
}