Cod sursa(job #1227005)

Utilizator TudorMTudor Moldovanu TudorM Data 9 septembrie 2014 12:10:01
Problema Algoritmul Bellman-Ford Scor 15
Compilator cpp Status done
Runda Arhiva educationala Marime 1.03 kb
#include<fstream>
#include<queue>
#define INF 1000000
using namespace std;
ifstream f("bellmanford.in");
ofstream g("bellmanford.out");
int n, m, a[1501][1501], viz[1501];
queue <int> c;
void citire()
{
    int i, j, x, y, pret;
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=m;j++)a[i][j]=INF;
    }
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>pret;
        a[x][y]=pret;
    }
}
void bmf()
{
    int i, u, p, mini, poz;
    for(i=2;i<=n;i++)viz[i]=INF;
    c.push(1);
    while(!c.empty())
    {
        p=c.front();
        mini=INF;
        for(i=1;i<=n;i++)
        {
            if(a[p][i]<mini&&viz[i]>viz[p]+a[p][i])
            {
                mini=a[p][i];
                poz=i;
            }
        }
        if(mini==INF)c.pop();
        else
        {
            viz[poz]=viz[p]+a[p][poz];
            c.push(poz);
        }
    }
}
int main()
{
    int i;
    citire();
    bmf();
    for(i=2;i<=n;i++)g<<viz[i]<<" ";
    f.close();
    g.close();
    return 0;
}