Cod sursa(job #2325308)

Utilizator Raresr14Rosca Rares Raresr14 Data 22 ianuarie 2019 14:32:21
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <fstream>
#include <climits>
#include <vector>
#include <set>
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
int i,n,m,x,y,z,D[50010],k;
vector<pair<int, int> > L[50010];
set<pair<int, int> > s;
int main(){
    fin>>n>>m;
    for(i=1;i<=m;i++){
        fin>>x>>y>>z;
        L[x].push_back(make_pair(y,z));
    }
    D[1]=0;
    for(i=2;i<=n;i++)
        D[i]=INT_MAX;
    s.insert(make_pair(0,1));
    while(!s.empty()){
        k=s.begin()->second;
        s.erase(s.begin());
        for(i=0;i<L[k].size();i++){
            int vec=L[k][i].first;
            int cost=L[k][i].second;
            if(D[vec]>D[k]+cost){
                s.erase(make_pair(D[vec],vec));
                D[vec]=D[k]+cost;
                s.insert(make_pair(D[vec],vec));
            }
        }
    }
    for(i=2;i<=n;i++)
        if(D[i]==INT_MAX)
            fout<<"0 ";
        else
            fout<<D[i]<<" ";
    return 0;
}