Cod sursa(job #1236141)

Utilizator TudorMTudor Moldovanu TudorM Data 1 octombrie 2014 15:43:01
Problema Drumuri minime Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.28 kb
#include<vector>
#include<fstream>
#include<queue>
#define INF 100000000000
using namespace std;
ifstream f("dmin.in");
ofstream g("dmin.out");
int n, m, nr[1501];
long long d[1501];
typedef struct
{
    int y, cost;
}MUCHIE;
vector <MUCHIE> v[1501];
queue <int> c;
MUCHIE w;
void citire()
{
    int i, j, x, y, pret;
    f>>n>>m;
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>pret;
        w.y=y;
        w.cost=pret;
        v[x].push_back(w);
        w.y=x;
        v[y].push_back(w);
    }
}
void bmf()
{
    int i, poz, p, ok1=0;
    for(i=2;i<=n;i++)d[i]=INF;
    d[1]=1;
    c.push(1);
    while(!c.empty())
    {
        p=c.front();
        ok1=0;
        for(i=0;i<v[p].size();i++)
        {
            poz=v[p][i].y;
            if(d[poz]>d[p]*v[p][i].cost)
            {
                ok1=1;
                d[poz]=d[p]*v[p][i].cost;
                c.push(poz);
                nr[poz]=1;
            }
            else if(d[poz]==d[p]*v[p][i].cost)
            {
                ok1=1;
                nr[poz]++;
                c.push(poz);
            }
        }
        c.pop();
    }
}
int main()
{
    int i;
    citire();
    bmf();
    for(i=2;i<=n;i++)g<<nr[i]<<" ";
    f.close();
    g.close();
    return 0;
}