Cod sursa(job #1124600)

Utilizator tudgal1001Profir Tudor tudgal1001 Data 26 februarie 2014 12:54:01
Problema Algoritmul Bellman-Ford Scor 35
Compilator cpp Status done
Runda Arhiva educationala Marime 1.32 kb
#include<cstdio>
#include<vector>
#include<queue>
#define NMax 50005
#define inf 1e9
using namespace std;

struct muchie
{ int nd,cost; };
muchie y;
int n,m,inqueue[NMax];
int d[NMax];

vector <muchie> v[NMax];
queue <int> q;

void citire ()
{
    int x,i;
    scanf("%d%d",&n,&m);
    for (i=1; i<=m; i++)
    {
        scanf("%d%d%d",&x,&y.nd,&y.cost);
        v[x].push_back(y);
    }
}

void init ()
{
    int i;
    for (i=2; i<=n; i++)
        d[i]=inf;
    d[1]=0;
}

int bellman_ford ()
{
    init();
    q.push(1);
    inqueue[1]=1;
    while (!q.empty())
    {
        int nod=q.front();
        q.pop();
        inqueue[nod]=0;
        for (int i=0; i<v[nod].size(); i++)
            if (d[v[nod][i].nd]>d[nod]+v[nod][i].cost)
            {
                d[v[nod][i].nd]=d[nod]+v[nod][i].cost;
                if (!inqueue[v[nod][i].nd])
                {
                    q.push(v[nod][i].nd);
                    inqueue[v[nod][i].nd]=1;
                }
            }
    }
    return 1;
}

int main ()
{
    freopen("bellmanford.in","r",stdin);
    freopen("bellmanford.out","w",stdout);
    citire();
    if (bellman_ford())
        for (int i=2; i<=n; i++)
            if (d[i]==inf)
                printf("0 ");
            else printf("%d ",d[i]);
    else
        printf("Ciclu negativ!");
    printf("\n");
    return 0;
}