Cod sursa(job #603174)

Utilizator a_h1926Heidelbacher Andrei a_h1926 Data 14 iulie 2011 22:39:22
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 2.2 kb
#include <iostream>
#include <vector>

#define Infinit 2000000000
#define NMax 50005
#define v first
#define c second

using namespace std;

vector < pair <int, int> > G[NMax];
int N, H[NMax], NH, Poz[NMax], D[NMax];

void Read ()
{
    freopen ("dijkstra.in", "r", stdin);
    int M;
    scanf ("%d %d", &N, &M);
    for (; M>0; --M)
    {
        int X, Y, C;
        scanf ("%d %d %d", &X, &Y, &C);
        G[X].push_back (make_pair (Y, C));
    }
}

void Print ()
{
    freopen ("dijkstra.out", "w", stdout);
    for (int i=2; i<=N; ++i)
    {
        if (D[i]==Infinit)
        {
            printf ("0 ");
        }
        else
        {
            printf ("%d ", D[i]);
        }
    }
}

inline void Swap (int X, int Y)
{
    int Aux;
    Aux=Poz[H[X]];
    Poz[H[X]]=Poz[H[Y]];
    Poz[H[Y]]=Aux;
    Aux=H[X];
    H[X]=H[Y];
    H[Y]=Aux;
}

void Percolate (int X)
{
    int Father=X/2;
    while (Father>0)
    {
        if (D[H[X]]<D[H[Father]])
        {
            Swap (X, Father);
            X=Father;
            Father=X/2;
        }
        else
        {
            return;
        }
    }
}

void Sift (int X)
{
    int Son=2*X;
    while (Son<=NH)
    {
        if (Son+1<=NH && D[H[Son+1]]<D[H[Son]])
        {
            ++Son;
        }
        if (D[H[X]]>D[H[Son]])
        {
            Swap (X, Son);
            X=Son;
            Son=2*X;
        }
        else
        {
            return;
        }
    }
}

void Delete (int X)
{
    Swap (X, NH);
    --NH;
    Sift (X);
}

void Initialize (int Start)
{
    NH=N;
    for (int i=1; i<=N; ++i)
    {
        D[i]=Infinit;
        H[i]=Poz[i]=i;
    }
    D[Start]=0;
    Swap (1, Start);
}

void Dijkstra (int Start)
{
    Initialize (Start);
    while (NH>0)
    {
        int X=H[1];
        Delete (1);
        for (unsigned i=0; i<G[X].size (); ++i)
        {
            int V=G[X][i].v, C=G[X][i].c;
            if (D[X]+C<D[V])
            {
                D[V]=D[X]+C;
                Percolate (Poz[V]);
            }
        }
    }
}

int main()
{
    Read ();
    Dijkstra (1);
    Print ();
    return 0;
}