Pagini recente » Cod sursa (job #5154) | Cod sursa (job #2365788) | Cod sursa (job #1915245) | Cod sursa (job #196946) | Cod sursa (job #2964928)
#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;
}