Cod sursa(job #2578323)

Utilizator GarvanGrachiIvan Valentinov GarvanGrachi Data 10 martie 2020 20:59:14
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.22 kb
#include<bits/stdc++.h>
#include<fstream>
using namespace std;

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

const int N = 100001;
const int oo = 1 << 25;

vector<pair<int,int > > g[N];
int n, m, f;
int dist[N];

void Dijkstra(int f){
    priority_queue<pair<int, int>, vector<pair<int,int> >, greater<pair<int, int> > > pq;
    pq.push({0, f});
    fill(dist, dist + n , oo);
    dist[f] = 0;
    while(!pq.empty()){
        //cout<<pq.top().first<<"\n\n";
        int u = pq.top().second;
        //if(pq.top().first>dist[u]){pq.pop(); continue;} ///optimzizaciq #1
        pq.pop();
        for(auto v : g[u]){
            int  d = v.second;
            int to = v.first;
            if(dist[to] > dist[u] + d){
                dist[to] = dist[u] + d;
                pq.push({dist[to], to});
            }
        }
    }
}

int main(){
    in >> n>> m;
    for(int i = 0 ; i < m ; ++ i){
        int u,v,w;
        in >> u >> v >> w;
        g[--u].push_back({--v, w});
        g[v].push_back({u, w});
    }
    //cin >> f;
    //cout << '\n';
    Dijkstra(0);
    for(int i=1;i<n;++i){
        if(dist[i]==oo) out<<"0 ";
        else out<<dist[i]<<' ';
    }
    return 0;
}