Pagini recente » Cod sursa (job #1533252) | Cod sursa (job #1847030) | Cod sursa (job #2097670) | Cod sursa (job #510496) | Cod sursa (job #144972)
Cod sursa(job #144972)
#include <stdio.h>
#include <vector>
#include <string.h>
const int inf = 1 << 30;
using namespace std;
int d[50001], h[50001], ph[50001], n, m;
void riseup(int);
void godown(int);
int main()
{
freopen("dijkstra.in", "r", stdin);
freopen("dijkstra.out", "w", stdout);
int i, j, a, b, c, sz, g[50001], x[50001], y[50001], z[50001];
int *v[50001], *cost[50001];
memset(g, 0, sizeof(g));
scanf("%d %d", &n, &m);
for(i = 1; i <= m; ++i)
{
scanf("%d %d %d", &x[i], &y[i], &z[i]);
++g[x[i]];
}
for(i = 1; i <= n; ++i)
{
v[i] = new int[g[i] + 2];
cost[i] = new int[g[i] + 2];
}
for(i = 1; i <= m; ++i)
{
v[x[i]][++v[x[i]][0]] = y[i];
cost[x[i]][v[x[i]][0]] = z[i];
}
for(i = 1; i <= n; ++i)
{
d[i] = inf;
h[i] = ph[i] = i;
}
d[1] = 0;
h[0] = n;
while(h[0])
{
for(i = 1; i <= g[h[1]]; ++i)
{
if(d[v[h[1]][i]] > d[h[1]] + cost[h[1]][i])
{
d[v[h[1]][i]] = d[h[1]] + cost[h[1]][i];
riseup(ph[v[h[1]][i]]);
}
}
h[1] = h[h[0]];
ph[h[1]] = 1;
--h[0];
godown(1);
}
for(i = 2; i <= n; ++i)
{
if(d[i] == inf)
d[i] = 0;
printf("%d ", d[i]);
}
return 0;
}
void godown(int pos)
{
int l = pos << 1, r = l + 1, best = pos;
if(l <= h[0] && d[h[l]] < d[h[best]])
{
best = l;
}
if(r <= h[0] && d[h[r]] < d[h[best]])
{
best = r;
}
if(best != pos)
{
int temp;
temp = h[best];
h[best] = h[pos];
h[pos] = temp;
ph[h[pos]] = pos;
ph[h[best]] = best;
godown(best);
}
}
void riseup(int pos)
{
int dad = pos >> 1;;
if(dad)
{
if(d[h[pos]] < d[h[dad]])
{
int temp;
temp = h[pos];
h[pos] = h[dad];
h[dad] = temp;
ph[h[pos]] = pos;
ph[h[dad]] = dad;
riseup(dad);
}
}
}