Cod sursa(job #2716344)

Utilizator Alex_tz307Lorintz Alexandru Alex_tz307 Data 5 martie 2021 08:35:02
Problema Drumuri minime Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.52 kb
#include <bits/stdc++.h>
#define ABS(x) ((x) >= 0 ? (x) : -(x))
#define INF 0x3f3f3f3f

using namespace std;
using ld = long double;

ifstream fin("dmin.in");
ofstream fout("dmin.out");

const int mod = 104659;
const int NMAX = 2048;
const ld EPS = 1e-6;
int N, M;
pair<ld,int> dp[NMAX];
vector<pair<int,ld>> G[NMAX];
priority_queue<pair<ld,int>> Q;

void add_self(int &a, int b) {
    a += b;
    if(a >= mod)
        a -= mod;
}

void Dijkstra() {
    for(int u = 1; u <= N; ++u)
        dp[u].first = INF;
    dp[1] = make_pair(0, 1);
    Q.emplace(0.0, 1);
    while(!Q.empty()) {
        int u = Q.top().second,
            d = -Q.top().first;
        Q.pop();
        if(d != dp[u].first)
            continue;
        for(const auto &v : G[u])
            if((dp[v.first].first - (dp[u].first + v.second)) > EPS) {
                dp[v.first].first = dp[u].first + v.second;
                dp[v.first].second = dp[u].second;
                Q.emplace(-dp[v.first].first, v.first);
            }
            else
                if(ABS(dp[v.first].first - (dp[u].first+ v.second)) <= EPS)
                    add_self(dp[v.first].second, dp[u].second);
    }
}

int main() {
    fin >> N >> M;
    for(int i = 0; i < M; ++i) {
        int u, v;
        ld w;
        fin >> u >> v >> w;
        w = log2(w);
        G[u].emplace_back(v, w);
        G[v].emplace_back(u, w);
    }
    Dijkstra();
    for(int u = 2; u <= N; ++u)
        fout << dp[u].second << ' ';
    fout << '\n';
}