Cod sursa(job #1929204)

Utilizator Horia14Horia Banciu Horia14 Data 17 martie 2017 11:56:01
Problema Algoritmul Bellman-Ford Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.75 kb
#include<cstdio>
#include<queue>
#define N_MAX 50001
#define INF 0x7fffffff
using namespace std;

int d[N_MAX], ItNod[N_MAX], n, m;
bool used[N_MAX];

struct nod
{
    int val, cost;
    nod *next;
};

nod *L[N_MAX];

void Insert(int x, int y, int c)
{
    nod *elem = new nod;
    elem->val = y;
    elem->cost = c;
    elem->next = L[x];
    L[x] = elem;
}

void Read()
{
    int i, a, b, c;
    FILE *fin = fopen("bellmanford.in","r");
    fscanf(fin,"%d%d",&n,&m);
    for(i=1; i<=m; i++)
    {
        fscanf(fin,"%d%d%d",&a,&b,&c);
        Insert(a,b,c);
    }
    fclose(fin);
}

int BellmanFord(int x)
{
    queue<int>Q;
    int i, Nod;
    bool ok;
    nod *p;
    for(i=1; i<=n; i++)
        d[i] = INF;
    d[x] = 0;
    Q.push(x);
    used[x] = true;
    ok = false;
    while(!Q.empty() && !ok)
    {
        Nod = Q.front();
        Q.pop();
        used[Nod] = false;
        p = L[Nod];
        while(p != NULL)
        {
            if(d[p->val] > d[Nod] + p->cost)
            {
                d[p->val] = d[Nod] + p->cost;
                if(!used[p->val])
                {
                    Q.push(p->val);
                    used[p->val] = true;
                    if(++ItNod[p->val] > n)
                        ok = true;
                }
            }
            p = p->next;
        }
    }
    if(ok)
        return 1;
    else return 0;
}

void Write()
{
    FILE *fout = fopen("bellmanford.out","w");
    if(BellmanFord(1))
        fprintf(fout,"Ciclu negativ!\n");
    else
    {
        for(int i=2; i<=n; i++)
            fprintf(fout,"%d ",d[i]);
        fprintf(fout,"\n");
    }
    fclose(fout);
}

int main()
{
    Read();
    Write();
    return 0;
}