Cod sursa(job #2377199)

Utilizator HoratioHoratiu Duma Horatio Data 8 martie 2019 23:15:10
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.34 kb
#include <cstdio>
#include <set>
#include <vector>
#include <bitset>
#define NMAX 50005
#define INF 1<<30
using namespace std;



vector<pair<int,int>> G[NMAX];

int dist[NMAX];

int n,m;

void citire()
{
    scanf("%d %d\n",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        scanf("%d %d %d\n",&a,&b,&c);
        G[a].push_back(make_pair(b,c));
    }
}


set<pair<int,int>> Q;

int Dijkstra()
{
    Q.insert(make_pair(0,1));
    while(!Q.empty())
    {
        int nod=Q.begin()->second;
        Q.erase(*Q.begin());
        for(auto it:G[nod])
        {
            int catre=it.first;
            int cost=it.second;


            if( dist[catre] > dist[nod] + cost)
            {
                if(dist[catre]!=INF)
                {
                    Q.erase(make_pair(dist[catre],catre));
                }
                dist[catre]=dist[nod]+cost;
                Q.insert(make_pair(dist[catre],catre));
            }
        }
    }

    for(int i=2;i<=n;i++)
        {
        if(dist[i]==INF)
            dist[i]=0;


        printf("%d ",dist[i]);
        }

}

int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    citire();

    for(int i=1;i<=n;i++)
        dist[i]=INF;

    dist[1]=0;

    Dijkstra();
    return 0;
}