Cod sursa(job #1793481)

Utilizator PaulStighiStiegelbauer Paul-Alexandru PaulStighi Data 31 octombrie 2016 08:30:14
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.84 kb
#include<fstream>
#include<vector>
#define oo 1<<30
#define NMax 50005
using namespace std;

ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");

int N,M,NHeap;
int D[NMax],Heap[NMax],Poz[NMax];

vector < pair <int,int> > G[NMax];

void Read()
{
    fin>>N>>M;

    for(int i = 1 ; i <= M ; ++i)
    {
        int x,y,c;  fin>>x>>y>>c;
        G[x].push_back(make_pair(y,c));
    }
}

void UpHeap(int Nod)
{
    int Tata = Nod / 2;

    if(Tata > 0 && D[Heap[Tata]] > D[Heap[Nod]])
    {
        swap(Heap[Tata],Heap[Nod]);
        swap(Poz[Heap[Tata]],Poz[Heap[Nod]]);
        UpHeap(Tata);
    }
}

void DownHeap(int Nod)
{
    int Son = Nod * 2;

    if(Son + 1 <= NHeap && D[Heap[Son+1]] < D[Heap[Son]])
        Son++;
    if(Son <= NHeap && D[Heap[Son]] < D[Heap[Nod]])
    {
        swap(Heap[Son],Heap[Nod]);
        swap(Poz[Heap[Son]],Poz[Heap[Son]]);
        DownHeap(Son);
    }
}

void Solve()
{
    for(int i = 1 ; i <= N ; ++i)
    {
        D[i] = oo;
        Heap[i] = i;
        Poz[i] = i;
    }

    D[1] = 0;
    NHeap = N;

    for(int k = 1 ; k <= N ; ++k)
    {
        int Nod = Heap[1];

        Heap[1] = Heap[NHeap];
        swap(Poz[Heap[1]],Poz[Heap[NHeap]]);
        NHeap--;

        for(int i = 0 ; i < (int) G[Nod].size() ; ++i)
        {
            int Vecin = G[Nod][i].first;
            int Cost = G[Nod][i].second;

            if(D[Nod] + Cost < D[Vecin])
            {
                D[Vecin] = D[Nod] + Cost;
                UpHeap(Poz[Vecin]);
            }
        }
    }
}

void Print()
{
    for(int i = 2 ; i <= N ; ++i)
    {
        if(D[i] == oo)  fout<<"0 ";
        else    fout<<D[i]<<" ";
    }
    fout<<"\n";
}

int main()
{
    Read();
    Solve();
    Print();

    fin.close();
    fout.close();
    return 0;
}