Cod sursa(job #2575748)

Utilizator anamariatoaderAna Toader anamariatoader Data 6 martie 2020 15:15:16
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.22 kb
#include <fstream>
#include <vector>
#include <queue>
#define inf 2000000000

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

int x,y,c,i,n,m,d[50005];
bool viz[50005];
struct arc{
    int nod,cost;
};
vector <arc> g[50005];
struct cmp{
    bool operator()(int &a, int &b){
        return d[a]>d[b];
    }
};
priority_queue <int, vector<int>, cmp> h;

void dijkstra(){
    while(!h.empty()){
        int nod=h.top();
        h.pop();
        if(!viz[nod])
            continue;
        viz[nod]=0;
        for(int i=0;i<g[nod].size();i++){
            int nou=g[nod][i].nod;
            int cost=g[nod][i].cost;
            if(d[nou]>d[nod]+cost){
                d[nou]=d[nod]+cost;
                if(!viz[nou]){
                    h.push(nou);
                    viz[nou]=1;
                }
            }
        }
    }
}

int main(){
    fin>>n>>m;
    for(i=1;i<=m;i++){
        fin>>x>>y>>c;
        g[x].push_back({y,c});
    }
    for(i=2;i<=n;i++)
        d[i]=inf;
    h.push(1);
    viz[1]=1;
    dijkstra();
    for(i=2;i<=n;i++)
        if(d[i]==inf)
            fout<<0<<' ';
        else
            fout<<d[i]<<' ';
    return 0;
}