Cod sursa(job #2117689)

Utilizator dragos231456Neghina Dragos dragos231456 Data 29 ianuarie 2018 10:12:17
Problema Algoritmul Bellman-Ford Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.26 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#define pb push_back
#define mp make_pair
/// http://www.spoj.com/problems/ARBITRAG/
using namespace std;
ifstream f("bellmanford.in");
ofstream g("bellmanford.out");
int n,v[50005],m,a,b,c,start,finish,cost,cycle[50005],inf=(1<<30);
vector< pair<int,int> >vecini[50005];
queue<int> q; bool ok=true;

void afisare()
{
    for(int i=1;i<=n;++i) cout<<v[i]<<' '; cout<<endl;
}

void BF(int x)
{
    q.push(x);
    while(!q.empty())
    {
        start=q.front(); q.pop();
        cycle[start]++;
        if(cycle[start]==n)
        {
            g<<"Ciclu negativ!"; ok=false; return;
        }
        for(int i=0;i<vecini[start].size();++i)
        {
            finish=vecini[start][i].first;
            cost=vecini[start][i].second+v[start];
            if(cost<v[finish])
            {
                v[finish]=cost;
                q.push(finish);
                //afisare();
            }
        }
    }
}

int main()
{
    f>>n>>m;
    for(int i=1;i<=m;++i)
    {
        f>>a>>b>>c;
        vecini[a].pb(mp(b,c));
    }
    for(int i=2;i<=n;++i) v[i]=inf;
    BF(1);
    if(ok)
    {
        for(int i=2;i<=n;++i) g<<v[i]<<' ';
    }
    return 0;
}