Cod sursa(job #2059595)

Utilizator cipri321Marin Ciprian cipri321 Data 7 noiembrie 2017 11:02:08
Problema Drumuri minime Scor 15
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.39 kb
#include <fstream>
#include <cmath>
#include <vector>
#include <set>
#define DIM 1501
#define e 0.00000001
#define INF 10000000.0
#define MOD 104659
using namespace std;
ifstream fi("dmin.in");
ofstream fo("dmin.out");
int n,m;
vector<pair<int,int> > V[DIM];
set<pair<int,int> > S;
double D[DIM];
int P[DIM];
void dijkstra()
{
    S.insert(make_pair(D[1],1));
    while(!S.empty())
    {
        pair<int,int>aux=(*S.begin());
        int nod=aux.second;
        S.erase(S.begin());
        for(auto it:V[nod])
        {
            int to=it.first;
            double cost=log10(it.second);
            if(D[nod]+cost<D[to]+e)
            {
                if(abs(D[to]-INF)>e)
                    S.erase(S.find(make_pair(D[to],to)));
                D[to]=D[nod]+cost;
                P[to]=(P[to]+P[nod])%MOD;
                S.insert(make_pair(D[to],to));
            }
            else
                if(abs(D[nod]+cost-D[to])<e)
                    P[to]=(P[to]+P[nod])%MOD;
        }
    }
}
int main()
{
    fi>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        fi>>a>>b>>c;
        V[a].push_back(make_pair(b,c));
        V[b].push_back(make_pair(a,c));
    }
    for(int i=1;i<=n;i++)
        D[i]=INF;
    D[1]=0,P[1]=1;
    dijkstra();
    for(int i=2;i<=n;i++)
        fo<<P[i]<<" ";
    fi.close();
    fo.close();
    return 0;
}