Cod sursa(job #1299611)

Utilizator witselWitsel Andrei witsel Data 23 decembrie 2014 19:07:32
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.24 kb
#include <iostream>
#include <cstdio>
#include <fstream>
const int maxn=50001;
const int inf=1<<30;
using namespace std;
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
struct graf
{
    int nod,cost;
    graf *next;
};
int n,m;
graf *a[maxn];
int d[maxn],q[maxn];
void add(int where, int what, int cost)
{
    graf *q=new graf;
    q->nod=what;
    q->cost=cost;
    q->next=a[where];
    a[where]=q;
}
void read()
{
    fin>>n>>m;
    int x,y,z;
    for(int i=1;i<=n;i++)
    {
        fin>>x>>y>>z;
        add(x,y,z);
    }

}
void dijkstra()
{
    for(int i=2;i<=n;i++)
        d[i]=inf;
    int mini,pmin=0;

    for(int i=1;i<=n;i++)
    {

        mini=inf;
        for(int j=1;j<=n;j++)
            if(d[j]<mini && !q[j])
                {
                    mini=d[j];
                    pmin=j;
                }
        q[pmin]=1;
        graf *t=a[pmin];
        while(t)
        {
            if(d[t->nod]>d[pmin]+t->cost)
                d[t->nod]=d[pmin]+t->cost;
            t=t->next;
        }
    }
}
int main()
{
    read();
    dijkstra();
    for(int i=2;i<=n;i++)
        if(d[i]==inf) fout<<0<<" ";
        else fout<<d[i]<<" ";
    fout<<" ";
    return 0;
}