Cod sursa(job #1560168)

Utilizator VladTiberiuMihailescu Vlad Tiberiu VladTiberiu Data 1 ianuarie 2016 21:16:08
Problema Algoritmul Bellman-Ford Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.25 kb
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>

#define NMax 50005
#define INF 0x3f3f3f3f
using namespace std;

int dist[NMax],ciclu[NMax];
int n,m,x,y,c;
queue<int> q;
vector<pair<int, int> > vecini[NMax];

int bellmanfordBFS(int x0){
    memset(dist,INF,sizeof(dist));

    dist[x0] = 0;
    q.push(x0);

    while(!q.empty()){
        int nod = q.front();
        q.pop();
        ciclu[nod] ++;
        if(ciclu[nod] > n){
            return -1;
        }
        for(int i = 0; i < vecini[nod].size(); ++i){
            if(dist[vecini[nod][i].first] > dist[nod] + vecini[nod][i].second){
                dist[vecini[nod][i].first] = dist[nod] + vecini[nod][i].second;
                q.push(vecini[nod][i].first);
            }
        }
    }
    return 1;
}
int main()
{
    freopen("bellmanford.in","r",stdin);
    freopen("bellmanford.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i = 1; i <= m; i++){
        scanf("%d%d%d",&x,&y,&c);
        vecini[x].push_back(make_pair(y,c));
    }
    bellmanfordBFS(1);
    if(bellmanfordBFS(1) > 0){
        for(int i = 2; i <= n; i++){
            printf("%d ",dist[i]);
        }
    }else
        printf("Ciclu negativ!");
    return 0;
}