Cod sursa(job #2377071)

Utilizator vladvaculinVlad V vladvaculin Data 8 martie 2019 21:27:34
Problema Algoritmul lui Dijkstra Scor 60
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.15 kb
#include <bits/stdc++.h>
using namespace std;

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

const int inf = (1<<31)-1;

vector <pair<int,int> > g[50002];
int dist[50002];

int n,m;

struct compare{
    bool operator()(int x, int y){
        dist[x] > dist[y];
    }
};
priority_queue <int, vector<int>, compare> q;
void djk(int start){

    for(int i = 2; i<=n; i++)
        dist[i] = inf;
    q.push(start);
    while(!q.empty()){
        int current = q.top();
        q.pop();

        for(unsigned int i = 0; i<g[current].size(); i++){
            int nb = g[current][i].first;
            int c = g[current][i].second;

            if(dist[current] + c < dist[nb]){
                dist[nb] = dist[current] + c;
                q.push(nb);
            }
        }
    }

}

int main(){
    fin >> n>>m;
    for(int i = 1; i<=m; i++){
        int x,y,c;
        fin>>x>>y>>c;
        g[x].push_back(make_pair(y,c));
    }

    djk(1);
    for(int i = 2; i <= n; i++){
        if(dist[i] != inf)
            fout << dist[i] <<" ";
        else
            fout << "0 ";
    }



    return 0;
}