Cod sursa(job #1264778)

Utilizator Adrian1997Radulescu Adrian Adrian1997 Data 16 noiembrie 2014 11:15:34
Problema Algoritmul Bellman-Ford Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.16 kb
#include <fstream>
#include <list>
#include <queue>
#include <bitset>
#define DIM 50011
#define INF 25000011
using namespace std;
ifstream f("bellmanford.in");
ofstream g("bellmanford.out");
int n,m,ng,nr;
int D[DIM],U[DIM];

list<pair<int,int> > L[DIM];
queue<int> q;
bitset<DIM> viz;

int main(void){
    register int i,j,x,y,c,nod;
    list<pair<int,int> >::iterator it;

    f>>n>>m;
    for(i=1;i<=m;i++){
        f>>x>>y>>c;
        L[x].push_back(make_pair(c,y));
    }
    for(i=2;i<=n;i++) D[i]=INF;

    q.push(1);
    while(!q.empty() && !ng){
        nod=q.front(),q.pop();
        for(it=L[nod].begin();it!=L[nod].end();it++){
            if(D[it->second]>D[nod]+it->first){
                U[it->second]++;
                if(U[it->second]==n){
                    ng=1;
                    break;
                }
                D[it->second]=D[nod]+it->first;
                if(!viz[it->second])
                    q.push(it->second),viz[it->second]=1;
            }
        }
    }
    if(ng)
        g<<"Ciclu negativ!";
    else
        for(i=2;i<=n;i++) g<<D[i]<<" ";
    f.close();
    g.close();
    return 0;
}