Cod sursa(job #2417415)

Utilizator hiimsobaDicianu Ioan-Alexandru hiimsoba Data 29 aprilie 2019 18:24:06
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.52 kb
#include <bits/stdc++.h>

#define inf 1 << 30

std::vector<int> dijkstra(std::vector<std::vector<std::pair<int, int>>>& Graph, int startNode) {
    std::vector<int> costs(Graph.size(), inf);
    std::vector<bool> visited(Graph.size(), false);
    costs[0] = 0;
    std::priority_queue<std::pair<int, int>, std::vector<std::pair<int, int>>, std::greater<std::pair<int, int>>> queue;
    queue.push(std::make_pair(0, 0));
    while (!queue.empty()) {
        auto top = queue.top();
        queue.pop();
        int cost = top.first;
        int node = top.second;
        if (visited[node]) continue;
        visited[node] = true;
        for (auto& neighbor : Graph[node]) {
            if (costs[neighbor.first] > cost + neighbor.second) {
                costs[neighbor.first] = cost + neighbor.second;
                queue.push({costs[neighbor.first], neighbor.first});
            }
        }
    }
    return costs;
}

int main() {
    assert(freopen("dijkstra.in", "r", stdin));
    int V, E;
    assert(scanf("%d %d ", &V, &E) == 2);
    std::vector<std::vector<std::pair<int, int>>> Graph (V);
    int x, y, w;
    for (int i = 0; i < E; ++i) {
        assert(scanf("%d %d %d ", &x, &y, &w) == 3);
        Graph[--x].push_back(std::make_pair(--y, w));
    }
    fclose(stdin);
    auto res = dijkstra(Graph, 0);
    assert(freopen("dijkstra.out", "w", stdout));
    for (size_t i = 1; i < res.size(); i++) {
        printf("%d ", (res[i] == inf ? 0 : res[i]));
    }
    fclose(stdout);
    return 0;
}