Problem:
MathJax.Hub.Config({ tex2jax: { inlineMath: [["$","$"], ["\\(","\\)"]], processEscapes: true }});

Single Source Shortest Path II

For a given weighted graph $G = (V, E)$, find the shortest path from a source to each vertex. For each vertex $u$, print the total weight of edges on the shortest path from vertex $0$ to $u$.

Input

In the first line, an integer $n$ denoting the number of vertices in $G$ is given. In the following $n$ lines, adjacency lists for each vertex $u$ are respectively given in the following format:

$u$ $k$ $v_1$ $c_1$ $v_2$ $c_2$ ... $v_k$ $c_k$

Vertices in $G$ are named with IDs $0, 1, ..., n-1$. $u$ is ID of the target vertex and $k$ denotes its degree. $v_i (i = 1, 2, ... k)$ denote IDs of vertices adjacent to $u$ and $c_i$ denotes the weight of a directed edge connecting $u$ and $v_i$ (from $u$ to $v_i$).

Output

For each vertex, print its ID and the distance separated by a space character in a line respectively. Print in order of vertex IDs.

Constraints

$1 \leq n \leq 10,000$

$0 \leq c_i \leq 100,000$

$|E| < 500,000$

All vertices are reachable from vertex $0$

Sample Input 1

5
0 3 2 3 3 1 1 2
1 2 0 2 3 4
2 3 0 3 3 1 4 1
3 4 2 1 0 1 1 4 4 3
4 2 2 1 3 3

Sample Output 1

0 0
1 2
2 2
3 1
4 3

Reference

Introduction to Algorithms, Thomas H. Cormen, Charles E. Leiserson, Ronald L. Rivest, and Clifford Stein. The MIT Press.

Solution in C (judge verdict: Time Limit Exceeded)

#include <stdio.h>
#include <stdlib.h>

void solve(int **G, long **cost, int n, int u, long *d, int t) {
  int i,j;
  int *flag;
  j = 0;
  flag = (int*)malloc(sizeof(int) * n);
  for (i = 0; G[u][i] != -1; i++) {
    if(d[G[u][i]] == -1 || d[G[u][i]] > t+cost[u][i]) {
      flag[j] = i;
      j++;
      d[G[u][i]] = t+cost[u][i];
    }
  }
  for (i = 0; i < j; i++) {
    solve(G,cost,n,G[u][flag[i]],d,t+cost[u][flag[i]]);
  }
  free(flag);
}
void sort(long *A, int *B, int a, int b){
  if(b == a+1) return;
  int m = a+(b-a)/2;
  sort(A,B,a,m);
  sort(A,B,m,b);
  int i,j,t;
  long *newA;
  int *newB;
  newA = (long*)malloc(sizeof(long)*(b-a));
  newB = (int*)malloc(sizeof(int) * (b-a));
  if(b-a == 0) printf("a");
  i = 0;j = 0;t = 0;
  while(t < b-a) {
    if((a+i < m && A[a+i] < A[m+j]) || m+j == b) {
      newA[t] = A[a+i];
      newB[t] = B[a+i];
      i++;
    }
    else{
      newA[t] = A[m+j];
      newB[t] = B[m+j];
      j++;
    }
    t++;
  }
  for (i = 0; i < b-a; i++) {
    A[a+i] = newA[i];
    B[a+i] = newB[i];
  }
  free(newA);
  free(newB);
}

int main()
{
  int n;
  int i,j,u,k,v;
  int **G;
  long **cost,c;
  long *d;
  scanf("%d",&n);
  G = (int**)malloc(sizeof(int*) * n);
  cost = (long**)malloc(sizeof(long*) * n);
  for (i = 0; i < n; i++) {
    scanf("%d %d", &u, &k);
    G[u] = (int*)malloc(sizeof(int) * (k+1));
    cost[u] = (long*)malloc(sizeof(long) * k);
    for (j = 0; j < k; j++) {
      scanf("%d %ld", &v, &c);
      G[u][j] = v;
      cost[u][j] = c;
    }
    if(k != 0)sort(cost[u],G[u],0,k);
    G[u][k] = -1;
  }
  d = (long*)malloc(sizeof(long) * n);
  d[0] = 0;
  for (i = 1; i < n; i++) {
    d[i] = -1;
  }
  solve(G,cost,n,0,d,0);
  for (i = 0; i < n; i++) {
    printf("%d %ld\n", i, d[i]);
  }
  return 0;
}