Cod sursa(job #2403135)

Utilizator gabriel-mocioacaGabriel Mocioaca gabriel-mocioaca Data 11 aprilie 2019 11:57:35
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.01 kb
#include<bits/stdc++.h>
#define cin in
#define cout out

using namespace std;

typedef pair<int,int> iiPair;

ifstream in("dijkstra.in");
ofstream out("dijkstra.out");

void dijkstra(int src, int n, vector<vector<iiPair>> graph){
    vector<int> dist(n + 1,INT_MAX);
    priority_queue<iiPair,vector<iiPair>,greater<iiPair>> pq;

    pq.emplace(0,src);
    dist[src] = 0;

    while(!pq.empty()){
        int u = pq.top().second;
        pq.pop();

        for(auto it : graph[u]){
            int v = it.first;
            int w = it.second;

            if(dist[v] > dist[u] + w){
                dist[v] = dist[u] + w;
                pq.emplace(dist[v],v);
            }
        }
    }

    for(int i = 2; i <= n; ++i){
        cout << dist[i] << ' ';
    }
}

int main(){
    int n, m, x, y, w;

    cin >> n >> m;

    vector<vector<iiPair>> graph(n + 1);

    while(m--){
        cin >> x >> y >> w;
        graph[x].emplace_back(y,w);
    }

    dijkstra(1,n,graph);
}