Cod sursa(job #971648)

Utilizator gbi250Gabriela Moldovan gbi250 Data 9 iulie 2013 19:50:26
Problema Algoritmul Bellman-Ford Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.53 kb
#include <cstdio>
#include <vector>
#include <queue>
#include <climits>
#define SIZE 36002
using namespace std;
struct node{
    int vx, cost;
} a;
int n, m, k, i, x, y, z;
long long d[SIZE][2];
vector <node> v[SIZE];
queue <int> c;

int main()
{
    freopen("catun.in", "r", stdin);
    freopen("catun.out", "w", stdout);
    scanf("%d %d %d", &n, &m, &k);
    for(i=1;i<=n;++i)
        d[i][0]=LONG_LONG_MAX;
    for(i=1;i<=k;++i)
    {
        scanf("%d", &x);
        d[x][0]=0;
        d[x][1]=-x;
        c.push(x);
    }
    for(i=1;i<=m;++i)
    {
        scanf("%d %d %d", &x, &y, &z);
        a.cost=z; a.vx=y;
        v[x].push_back(a);
        a.vx=x;
        v[y].push_back(a);
    }
    while(!c.empty())
    {
        x=c.front(); c.pop();
        for(i=0;i<v[x].size();++i)
            if(d[v[x][i].vx][0]>d[x][0]+v[x][i].cost)
            {
                d[v[x][i].vx][0]=d[x][0]+v[x][i].cost;
                if(d[x][1]>0)
                    d[v[x][i].vx][1]=d[x][1];
                else
                    d[v[x][i].vx][1]=-d[x][1];
                c.push(v[x][i].vx);
            }
            else if(d[v[x][i].vx][0]==d[x][0]+v[x][i].cost && d[x][1]<d[v[x][i].vx][1])
            {
                if(d[x][1]>0)
                    d[v[x][i].vx][1]=d[x][1];
                else
                    d[v[x][i].vx][1]=-d[x][1];
                c.push(v[x][i].vx);
            }
    }
    for(i=1;i<=n;++i)
        if(d[i][0]!=INT_MAX && d[i][1]>0)
                printf("%lld ", d[i][1]);
        else printf("0 ");
    return 0;
}