Cod sursa(job #2812242)

Utilizator domistnSatnoianu Dominic Ioan domistn Data 4 decembrie 2021 11:28:09
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.15 kb
#include <iostream>
#include <set>
#include <vector>

#define NMAX 50005
#define INF ((1 << 30) - 1)

using namespace std;

typedef pair<int, int> pii;

int n, m, ans[NMAX];
vector<pii> adj[NMAX];
set<pii> st;

void dij(const int SN = 1) {
    ans[SN] = 0;
    st.insert({0, SN});
    while(!st.empty()) {
        const int crt = (*st.begin()).second;
        st.erase(st.begin());

        for(const auto &el : adj[crt]) {
            const int ncrt = el.first;
            if(ans[ncrt] > ans[crt] + el.second) {
                if(ans[ncrt] != INF)
                    st.erase(st.find({ans[ncrt], ncrt}));
                ans[ncrt] = ans[crt] + el.second;
                st.insert({ans[ncrt], ncrt});
            }
        }
    }
}

int main()
{
    freopen("dijkstra.in", "r", stdin);
    freopen("dijkstra.out", "w", stdout);
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i)
        ans[i] = INF;
    for(int x, y, z; m; --m) {
        scanf("%d%d%d", &x, &y, &z);
        adj[x].push_back({y, z});
    }
    dij();
    for(int i = 2; i <= n; ++i) printf("%d ", (ans[i] == INF ? 0 : ans[i]));
    return 0;
}