Cod sursa(job #1118388)

Utilizator cristitamasTamas Cristian cristitamas Data 24 februarie 2014 10:40:25
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.28 kb
#include <iostream>
#include <cstdio>
#include <queue>
#include <vector>
#define INF 0x3f3f3f3f
#define Nmax 50010
using namespace std;

vector <pair <int, int> > Graf[Nmax];
priority_queue <pair <int, int>, vector <pair <int, int> >, greater <pair <int, int> > > C;

int N,M,Cost[Nmax];

void Citire()
{
    int x,y,c;
    scanf("%d %d",&N,&M);
    for(int i=1;i<=M;++i)
    {
        scanf("%d %d %d",&x,&y,&c);
        Graf[x].push_back(make_pair(c,y));
    }
    for(int i=2;i<=N;++i)
        Cost[i]=INF;
}

void Rezolvare()
{
    int x;
    vector <pair <int, int> >::iterator it;
    C.push(make_pair(0,1));
    while(!C.empty())
    {
        x=C.top().second;
        C.pop();
        for(it=Graf[x].begin();it!=Graf[x].end();++it)
        {
            if(Cost[(*it).second]>Cost[x]+(*it).first)
            {
                Cost[(*it).second]=Cost[x]+(*it).first;
                C.push(make_pair(Cost[(*it).second],(*it).second));
            }
        }
    }
}



int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    Citire();
    Rezolvare();
    for(int i=2;i<=N;++i)
        if(Cost[i]!=INF)
            printf("%d ",Cost[i]);
        else
            printf("%d ",0);
    return 0;
}