Cod sursa(job #1689135)

Utilizator VladTiberiuMihailescu Vlad Tiberiu VladTiberiu Data 13 aprilie 2016 23:09:50
Problema Algoritmul Bellman-Ford Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.13 kb
#include <bits/stdc++.h>

#define NMax 50005
#define INF 0x3f3f3f3f
using namespace std;
ifstream f("bellmanford.in");
ofstream g("bellmanford.out");

int n,m,x,y,c;
vector<pair<int,int> > G[NMax];
int dist[NMax],v[NMax],viz[NMax];
queue<int> q;

int main()
{
    f >> n >> m;
    for(int i = 1; i <= m; ++i){
        f >> x >> y >> c;
        G[x].push_back(make_pair(y,c));
    }
    memset(dist,INF,sizeof(dist));
    q.push(1);
    dist[1] = 0;
    while(!q.empty()){
        int nod = q.front();
        viz[nod] = 0;
        v[nod]++;
        if(v[nod] > n){
            g << "Ciclu negativ!"<< '\n';
            return 0;
        }
        q.pop();
        for(int i = 0; i < G[nod].size(); ++i){
            if(dist[G[nod][i].first] > dist[nod] + G[nod][i].second){
                dist[G[nod][i].first] = dist[nod] + G[nod][i].second;

                if(!viz[G[nod][i].first]){
                    viz[G[nod][i].first] = 1;
                    q.push(G[nod][i].first);
                }
            }
        }
    }
    for(int i = 2; i <= n; ++i){
        g << dist[i] << ' ';
    }
    return 0;
}