Cod sursa(job #2567744)

Utilizator denmirceaBrasoveanu Mircea denmircea Data 3 martie 2020 18:40:13
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.92 kb
#include <bits/stdc++.h>
#define dim 100005
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
int d[dim],n,m,i;
set< pair<int,int> > s;
vector < pair<int,int> > L[dim];
void dijkstra(int nod){
    s.insert({0,nod});
    while(s.size()){
        nod=s.begin()->second;
        s.erase(s.begin());
        for(auto v:L[nod]){
            int vecin=v.first;
            int cost=v.second;
            if(d[vecin]>cost+d[nod]){
                s.erase({d[vecin],vecin});
                d[vecin]=cost+d[nod];
                s.insert({d[vecin],vecin});
            }
        }
    }
}
int main(){
    fin>>n>>m;
    for(i=1;i<=m;i++){
        int x,y,z;
        fin>>x>>y>>z;
        L[x].push_back({y,z});
    }
    memset(d,10,sizeof(d));
    int inf=d[1];
    d[1]=0;
    //cout<<inf<<endl;
    dijkstra(1);
    for(i=2;i<=n;i++)
        fout<< (d[i]!=inf?d[i]:0) <<" ";
}