Cod sursa(job #2115715)

Utilizator alex2kamebossPuscasu Alexandru alex2kameboss Data 27 ianuarie 2018 06:46:57
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.02 kb
#include <iostream>
#include <cstdio>
#include <vector>
#include <cstring>
#include <set>

using namespace std;
int INF=0x3f3f3f3f;
vector <pair<int,int>> g[50001];
int d[50001];
int n,m;
int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    scanf("%d %d", &n,&m);
    int a,b,c;
    for(int i=0;i<m;++i)
        scanf("\n%d %d %d", &a,&b,&c),
        g[a-1].push_back({b-1,c});
    memset(d,INF,sizeof d);
    set<pair<int,int>> q;
    q.insert({0,0});
    d[0]=0;
    while(!q.empty())
    {
        a=q.begin()->second;
        q.erase(q.begin());
        for(pair<int,int> i: g[a])
        {
            b=i.first;
            c=i.second;
            if(d[b]>d[a]+c)
            {
                if(d[b]!=INF)
                    q.erase(q.find({d[b],b}));
                d[b]=d[a]+c;
                q.insert({d[b],b});
            }
        }
    }
    for(int i=1;i<n;++i)
    {
        if(d[i]!=INF)
            printf("%d ", d[i]);
        else
            printf("0 ");
    }
    return 0;
}