Cod sursa(job #2069142)

Utilizator DavidDragulinDragulin David DavidDragulin Data 18 noiembrie 2017 12:08:52
Problema Algoritmul Bellman-Ford Scor 35
Compilator cpp Status done
Runda Arhiva educationala Marime 1.23 kb
#include <bits/stdc++.h>

#define mp make_pair
#define f first
#define s second
#define nmax 50009
#define pb push_back
#define inf INT_MAX
using namespace std;
ifstream fin("bellmanford.in");
ofstream fout("bellmanford.out");
priority_queue <pair<int,int> ,vector < pair <int,int> > ,greater <pair <int,int> > > h;
vector <pair <int,int> > v[nmax];
int n,m,a,b,c,d[nmax],i,nod;
bool pr[nmax];
int main()
{
    fin.sync_with_stdio(false);
    fin>>n>>m;
    for(i=1; i<=m; i++)
    {
        fin>>a>>b>>c;
        v[a].pb(mp(c,b));
    }
    h.push(mp(0,1));
    d[1]=0;
    for(i=2; i<=n; i++)
        d[i]=inf;
    while(!h.empty())
    {
        nod=h.top().s;
        h.pop();
            for(i=0; i<v[nod].size(); ++i)
            {
                if(d[nod]+v[nod][i].f<d[v[nod][i].s])
                {
                    d[v[nod][i].s]=d[nod]+v[nod][i].f;
                    h.push(mp(d[v[nod][i].s],v[nod][i].s));
                }
            }
            pr[nod]++;
        if(pr[nod]==n)
        {
            fout<<-1;
            return 0;
        }
    }
    for(i=2; i<=n; ++i)
        if(d[i]!=inf)
            fout<<d[i]<<" ";
        else
            fout<<0<<" ";
    return 0;
}