Cod sursa(job #903246)

Utilizator lianaliana tucar liana Data 1 martie 2013 19:26:25
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.25 kb
#include<stdio.h>
#include<vector>
#include<set>
using namespace std;
#define nmax 50005
#define inf nmax*1000
struct element{long n, c;};
long a, b, c, dmin[nmax], n, m, i;
vector <element> ma[nmax];
vector <element> ::iterator it;
set <pair <long, long> > h;
pair <long, long> pr;
element el;

void citire()
{
    scanf("%ld %ld",&n,&m);
    for (i=1;i<=m;i++)
    {
        scanf("%ld %ld %ld",&a,&b,&c);
        el.n=b; el.c=c; ma[a].push_back(el);
    }
}

void rezolvare()
{
    for (i=1;i<=n;i++)
        dmin[i]=inf;
    dmin[1]=0;
    h.insert(make_pair(0,1));
    while (h.size())
    {
        pr=*h.begin();  h.erase(h.begin());
        for (it=ma[pr.second].begin();it!=ma[pr.second].end();it++)
            if (dmin[(*it).n]>pr.first+(*it).c)
            {
                h.erase(make_pair(dmin[(*it).n],(*it).n));
                dmin[(*it).n]=pr.first+(*it).c;
                h.insert(make_pair(dmin[(*it).n],(*it).n));
            }
    }
}
int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    citire();
    rezolvare();
    for (i=2;i<=n;i++)
        if (dmin[i]!=inf)
            printf("%ld ",dmin[i]);
        else
            printf("0 ");
    return 0;
}