Cod sursa(job #1545798)

Utilizator andrei_diaconuAndrei Diaconu andrei_diaconu Data 7 decembrie 2015 08:05:33
Problema Drumuri minime Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.91 kb
#include <fstream>
#include <vector>
#include <queue>
#include <string.h>
#include <cmath>

#define NMax 1510
#define INF 2000000000
#define eps 1.e-10
#define MOD 104659

using namespace std;

ifstream f("dmin.in");
ofstream g("dmin.out");

int nodes, edges, roadsNum[NMax];
double distances[NMax];
vector < pair<int, double> > G[NMax];
queue <int> QU;
bool mark[NMax];

class cmp {
public:
    bool operator() (pair<int, double> d1, pair<int, double> d2)
    {
        return d1.second > d2.second;
    }
};
priority_queue< pair<int, double>, vector< pair<int, double> >, cmp > H;

void dijkstra(int node)
{
    roadsNum[node] = 1;
    mark[1] = true;

    for (int i = 2; i <= nodes; i++)
        distances[i] = INF;

    H.push(make_pair(node, 0));

    while (!H.empty()) {
        int crtNode = H.top().first;
        mark[crtNode] = true;
        H.pop();

        for (vector < pair<int, double> >::iterator it = G[crtNode].begin(); it != G[crtNode].end(); it++) {
            int q = it->first;
            if (!mark[it->first] && distances[it->first] - (distances[crtNode] + it->second) > eps) {
                distances[it->first] = distances[crtNode] + it->second;
                H.push(make_pair(it->first, distances[it->first]));
                roadsNum[it->first] = roadsNum[crtNode];
            }
            else if (!mark[it->first] && abs(distances[it->first] - (distances[crtNode] + it->second)) <= eps)
                roadsNum[it->first] = (roadsNum[it->first] + roadsNum[crtNode]) % MOD;
        }
    }
}

int main()
{
    f >> nodes >> edges;

    int n1, n2, c;
    for (int i = 1; i <= edges; i++) {
        f >> n1 >> n2 >> c;

        G[n1].push_back(make_pair(n2, log10(c)));
        G[n2].push_back(make_pair(n1, log10(c)));
    }

    dijkstra(1);


    for (int i = 2; i <= nodes; i++)
        g << roadsNum[i] << " ";

    return 0;
}