Cod sursa(job #2964928)

Utilizator SochuDarabaneanu Liviu Eugen Sochu Data 14 ianuarie 2023 10:06:35
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.22 kb
#include <bits/stdc++.h>
#define pii pair < int , int >
#define pb push_back

using namespace std;

const int N = 5e4 + 10;

int n , m;
int d[N];
vector < pii > G[N];

struct Heap
{
    int sz;
    int *heap , *pos;

    Heap(int n)
    {
        pos = new int[n + 10];
        heap = new int[n + 10];
        sz = 0;
    }

    void upg(int node)
    {
        while(node > 1)
        {
            if(d[heap[node]] < d[heap[node / 2]])
            {
                swap(d[heap[node]] , d[heap[node / 2]]);
                swap(pos[heap[node]] , pos[heap[node / 2]]);
                node /= 2;
            }
            else break;
        }
    }

    void dem(int node)
    {
        while(1)
        {
            int son = 2 * node;

            if(son + 1 <= sz && d[heap[son]] > d[heap[son + 1]])
                son++;

            if(son <= sz && d[heap[node]] > d[heap[son]])
            {
                swap(d[heap[node]] , d[heap[son]]);
                swap(pos[heap[node]] , pos[heap[son]]);
                node = son;
            }
            else break;
        }
    }

    void add(int x)
    {
        heap[++sz] = x;
        pos[x] = sz;
        upg(sz);
    }

    void rem()
    {
        pos[heap[1]] = 0;
        heap[1] = heap[sz];
        pos[heap[1]] = 1;
        --sz;
        dem(1);
    }
} H(N);


void dijkstra()
{
    for(int i = 1 ; i <= n ; i++)
        d[i] = INT_MAX;

    d[1] = 0;
    H.add(1);

    while(H.sz)
    {
        int node = H.heap[1];
        H.rem();

        for(auto i : G[node])
            if(d[i.first] > d[node] + i.second)
            {
                int p = H.pos[i.first];
                d[i.first] = d[node] + i.second;

                if(p) H.upg(p);
                else H.add(i.first);
            }
    }
}

int main()
{
    freopen("dijkstra.in" , "r" , stdin) , freopen("dijkstra.out" , "w" , stdout);

    cin >> n >> m;

    for(int i = 1 ; i <= m ; i++)
    {
        int x , y , c;
        cin >> x >> y >> c;
        G[x].pb({y , c});
        G[y].pb({x , c});
    }

    dijkstra();

    for(int i = 2 ; i <= n ; i++)
        cout << d[i] << ' ';

    return 0;
}