Cod sursa(job #1880654)

Utilizator AndreiGrigorasAndrei Grigoras AndreiGrigoras Data 15 februarie 2017 20:51:17
Problema Drumuri minime Scor 100
Compilator cpp Status done
Runda becreative17 Marime 1.98 kb
#include <fstream>
#include <vector>
#include <queue>
#include <utility>
#include <cmath>
#define NMAX 1600
#define MOD 104659
#define PHI 1e-8
#define INF 1e40
#define mp make_pair

using namespace std ;

ofstream fout ( "dmin.out" ) ;

int N , M , paths[NMAX] , inQueue[NMAX] ;
double dist[NMAX] ;
vector < pair < int , double > > TT[NMAX] ;
queue < int > Q ;


void read()
{
    int x , y , cost ;
    freopen ( "dmin.in" , "r" , stdin ) ;
    scanf ( "%d %d" , &N , &M ) ;
    for ( int i = 1 ; i <= M ; i++ )
    {
        scanf ( "%d %d %d" , &x , &y , &cost ) ; // logaritmam costul pentru a nu lucra cu numere mari si pentru ca log(a*b) = log a + log b
        TT[x].push_back( mp( y , log(cost) ) ) ;
        TT[y].push_back( mp( x , log(cost) ) ) ;
    }
    for ( int i = 2 ; i <= N ; i++ )
        dist[i] = INF ;
}

void bellman_ford()
{
    Q.push(1);
    inQueue[1] = 1;
    paths[1] = 1;

    while(!Q.empty())
    {
        int node = Q.front();
        Q.pop();
        vector<pair<int, double> >::iterator it = TT[node].begin();
        while(it != TT[node].end())
        {
            pair<int, double> neigh = *it;
            if(dist[neigh.first] - PHI > dist[node] + neigh.second)
            {
                dist[neigh.first] = dist[node] + neigh.second;
                paths[neigh.first] = paths[node];
                if(inQueue[neigh.first] == 0)
                {
                    Q.push(neigh.first);
                    inQueue[neigh.first] = 1;
                }
            }
            else
            {
                if(dist[neigh.first] + PHI >= dist[node] + neigh.second)
                {
                    paths[neigh.first] += paths[node];
                    paths[neigh.first] %= MOD ;
                }
            }

            it++;
        }

    }

    for ( int i = 2 ; i <= N ; i++ )
        fout << paths[i] << ' ' ;
}


int main()
{
    read() ;
    bellman_ford() ;
    return 0;
}