Cod sursa(job #2040312)

Utilizator Dinu2005Dinu I Dinu2005 Data 15 octombrie 2017 17:42:44
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.09 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
#define Nmax 50005
#define INF 1000000000

using namespace std;
int n;
priority_queue<pair<int,int>>q;
vector<pair<int,int>>L[Nmax];
vector<int>d;
inline void init(){
    for(int i=1;i<=n+1;i++){
        d.push_back(INF);
    }
}
void dijkstra(){
    init();
    q.push(make_pair(0,1));
    d[1]=0;
    while(!q.empty()){
        pair<int,int>p=q.top();
        q.pop();
        p.first=-p.first;
        for(auto it: L[p.second]){
            if(d[it.first]>it.second+p.first){
                d[it.first]=it.second+p.first;
                q.push(make_pair(-d[it.first],it.first));
            }
        }
    }
}
int main()
{
    ifstream cin("dijkstra.in");
    ofstream cout("dijkstra.out");
    int m,x,y,z;
    cin>>n>>m;
    for(int i=1;i<=m;++i){
        cin>>x>>y>>z;
        L[x].push_back(make_pair(y,z));
    }
    dijkstra();
    for(int i=2;i<=n;++i){
        if(d[i]==INF)
           cout<<0<<" ";
        else{
            cout<<d[i]<<" ";
        }
    }
    return 0;
}