Cod sursa(job #1816621)

Utilizator GeanaVladGeana Vlad GeanaVlad Data 26 noiembrie 2016 17:54:11
Problema Algoritmul Bellman-Ford Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.39 kb
#include <iostream>
#include<fstream>
#include<queue>
#define inf 1000000000
#define mp make_pair
#define f first
#define s second
using namespace std;
ifstream f("bellmanford.in");
ofstream gout("bellmanford.out");
int n,m,i,cst,v[50001],j,k,x,y,nod;
vector<int>q;
vector<pair<int,int> >g[50001];
bool ok,e[50001];
int functie(int i,int j,int c)
{
    if(v[j]>v[i]+c)
    {
        v[j]=v[i]+c;
        return 1;
    }
    return 0;
}
void sterge(int j)
{
    swap(q[j],q[q.size()-1]);
    q.pop_back();
}
int main()
{
    f>>n>>m;
    q.push_back(0);
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>cst;
        g[x].push_back(mp(y,cst));
    }
    v[1]=0;
    for(i=2;i<=n;i++)
        v[i]=inf;
    q.push_back(1);
    e[1]=1;
    for(i=1;i<n;i++)
        for(j=1;j<q.size();j++)
    {
        nod=q[j];
        for(k=0;k<g[nod].size();k++)
        if(functie(nod,g[nod][k].f,g[nod][k].s))
        {
            if(!e[g[nod][k].f])
            q.push_back(g[nod][k].f),e[g[nod][k].f]=1;
        }
        sterge(j);
        e[nod]=0;
    }
    ok=1;
    for(j=1;j<q.size();j++)
    {
        nod=q[j];
        for(k=0;k<g[nod].size();k++)
        if(functie(nod,g[nod][k].f,g[nod][k].s))
        {
            ok=0;
            gout<<"Ciclu negativ!";
        }
        sterge(j);
    }
    if(ok)
        for(i=2;i<=n;i++)
        gout<<v[i]<<' ';
}