Cod sursa(job #1197379)

Utilizator LurchssLaurentiu Duma Lurchss Data 11 iunie 2014 20:01:48
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.06 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
#include <string.h>

#define max 50005
#define pb push_back
#define inf 0x3f3f3f3f
using namespace std;
priority_queue <pair <int,int> > q;
vector <pair <int,int> > g[max];
int n,m;
int d[max];
void read()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=m;i++)
        {int a,b,c;
        scanf("%d %d %d",&a,&b,&c);
        g[a].pb(make_pair(c,b));
        }
}

void dijkstra()
{
    q.push(make_pair(0,1));
    memset(d,inf,sizeof(d));
    while(!q.empty())
    {
        int p=q.top().second;
        q.pop();
        for(vector <pair <int,int> >::iterator it=g[p].begin();it!=g[p].end();it++)
            if(d[it->second]>d[p]+it->first)
                {d[it->second]=d[p]+it->first;
                q.push(*it);}
    }
}
int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    read();
    dijkstra();
    for(int i=2;i<=n;i++)
        if(d[i]!=inf)
        printf("%d ",d[i]);
        else printf("0 ");
    return 0;
}