Cod sursa(job #556495)

Utilizator jupanubv92Popescu Marius jupanubv92 Data 16 martie 2011 10:14:22
Problema Algoritmul Bellman-Ford Scor 95
Compilator cpp Status done
Runda Arhiva educationala Marime 1.47 kb
#include<stdio.h>
#include<algorithm>
#include<queue>
#define INF 20000001
#define Nmx 50002

int n,m,viz[Nmx],cost[Nmx],cnt[Nmx];
int nr;
using namespace std;
queue<int>Q;

struct nod
{
    int inf;
    int cost;
    nod *urm;
};
nod *G[Nmx];

void add(int x,int y,int c)
{
    nod *aux=new nod;
    aux->inf=y;
    aux->cost=c;
    aux->urm=G[x];
    G[x]=aux;
}

void citire()
{
    int x,y,c;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;++i)
    {
        scanf("%d%d%d",&x,&y,&c);
        add(x,y,c);
    }
}


int solve()
{
    int st;
    for(int i=2;i<=n;++i)
        cost[i]=INF;
    viz[1]=1;
    Q.push(1);
    long long pas=0;
    while(!Q.empty()&&pas<=(long long)n*m)
    {
        pas++;
        st=Q.front();
        viz[st]=0;
        for(nod *p=G[st];p;p=p->urm)
            if(cost[st]+p->cost<cost[p->inf])
            {
                cost[p->inf]=cost[st]+p->cost;
                if(!viz[p->inf])
                {
                    viz[p->inf]=1;
                    Q.push(p->inf);
                    cnt[p->inf]++;
                }
                else if(cnt[p->inf]>n ) return 1;
            }
        Q.pop();
    }
    return 0;
}

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