Cod sursa(job #1597928)

Utilizator andreigasparoviciAndrei Gasparovici andreigasparovici Data 12 februarie 2016 14:51:37
Problema Algoritmul lui Dijkstra Scor 80
Compilator cpp Status done
Runda Arhiva educationala Marime 1.11 kb
#include <cstdio>
#include <vector>
#include <queue>
#include <map>
#include <functional>
#include <climits>
using namespace std;
int N,M;
vector<int>graf[50001];
map<pair<int,int>,int> cost;
void dijkstra(int stop)
{
    int *a=new int[N+1];
    for(int i=2;i<=N;i++)
        a[i]=INT_MAX;
    a[1]=0;
    priority_queue<int,vector<int>,greater<int> >q;
    q.push(1);
    while(!q.empty())
    {
        int nod=q.top();
        q.pop();
        for(int i=0;i<graf[nod].size();i++)
        {
            int v=graf[nod][i];
            if(a[nod]+cost[make_pair(nod,v)]<a[v])
            {
                a[v]=a[nod]+cost[make_pair(nod,v)];
                q.push(v);
            }
        }
    }
    for(int i=2;i<=N;i++)
        if(a[i]==INT_MAX)
            printf("0 ");
        else printf("%d ",a[i]);

}
int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    scanf("%d %d",&N,&M);
    for(int i=0;i<M;i++)
    {
        int from,to,c;
        scanf("%d %d %d",&from,&to,&c);
        graf[from].push_back(to);
        cost[make_pair(from,to)]=c;
    }
    dijkstra(2);
    return 0;
}