feat: add johnson algorithm for all pairs shortest paths

graph/johnson.ts

@@ -0,0 +1,52 @@
+import { bellmanFord } from './bellman_ford'
+import { dijkstra } from './dijkstra'
+
+/**
+ * @function johnson
+ * @description Compute the shortest path for all pairs of nodes. The input graph is in adjacency list form. It is a multidimensional array of edges. graph[i] holds the edges for the i'th node. Each edge is a 2-tuple where the 0'th item is the destination node, and the 1'th item is the edge weight. Returned undefined if the graph has negative weighted cycles.
+ * @Complexity_Analysis
+ * Time complexity: O(VElog(V))
+ * Space Complexity: O(V^2) to hold the result
+ * @param {[number, number][][]} graph - The graph in adjacency list form
+ * @return {number[][]} - A matrix holding the shortest path for each pair of nodes. matrix[i][j] holds the distance of the shortest path (i -> j).
+ * @see https://en.wikipedia.org/wiki/Johnson%27s_algorithm
+ */
+export const johnson = (graph: [number, number][][]): number[][] | undefined => {
+  let N = graph.length;
+
+  // Add a new node and 0 weighted edges from the new node to all existing nodes.
+  let newNodeGraph = structuredClone(graph);
+  let newNode: [number, number][] = [];
+  for (let i = 0; i < N; ++i) {
+    newNode.push([i, 0]);
+  }
+  newNodeGraph.push(newNode);
+
+  // Compute distances from the new node to existing nodes using the Bellman-Ford algorithm.
+  let adjustedGraph = bellmanFord(newNodeGraph, N);
+  if (adjustedGraph === undefined) {
+    // Found a negative weight cycle.
+    return undefined;
+  }
+
+  for (let i = 0; i < N; ++i) {
+    for (let edge of graph[i]) {
+      // Adjust edge weights using the Bellman Ford output weights. This ensure that:
+      // 1. Each weight is non-negative. This is required for the Dijkstra algorithm.
+      // 2. The shortest path from node i to node j consists of the same nodes with or without adjustment.
+      edge[1] += adjustedGraph[i] - adjustedGraph[edge[0]];
+    }
+  }
+
+  let shortestPaths: number[][] = [];
+  for (let i = 0; i < N; ++i) {
+    // Compute Dijkstra weights for each node and re-adjust weights to their original values.
+    let dijkstraShorestPaths = dijkstra(graph, i);
+    for (let j = 0; j < N; ++j) {
+      dijkstraShorestPaths[j] += adjustedGraph[j] - adjustedGraph[i];
+    }
+    shortestPaths.push(dijkstraShorestPaths);
+  }
+  return shortestPaths;
+}
+

graph/test/johnson.test.ts

@@ -0,0 +1,107 @@
+import { johnson } from "../johnson";
+
+describe("johnson", () => {
+
+  const init_graph = (N: number): [number, number][][] => {
+    let graph = Array(N);
+    for (let i = 0; i < N; ++i) {
+      graph[i] = [];
+    }
+    return graph;
+  }
+
+  const add_edge = (graph: [number, number][][], a: number, b: number, weight: number) => {
+    graph[a].push([b, weight]);
+    graph[b].push([a, weight]);
+  }
+
+  it("should return the correct value", () => {
+    let graph = init_graph(9);
+    add_edge(graph, 0, 1, 4);
+    add_edge(graph, 0, 7, 8);
+    add_edge(graph, 1, 2, 8);
+    add_edge(graph, 1, 7, 11);
+    add_edge(graph, 2, 3, 7);
+    add_edge(graph, 2, 5, 4);
+    add_edge(graph, 2, 8, 2);
+    add_edge(graph, 3, 4, 9);
+    add_edge(graph, 3, 5, 14);
+    add_edge(graph, 4, 5, 10);
+    add_edge(graph, 5, 6, 2);
+    add_edge(graph, 6, 7, 1);
+    add_edge(graph, 6, 8, 6);
+    add_edge(graph, 7, 8, 7);
+
+    let expected = [
+      [0, 4, 12, 19, 21, 11, 9, 8, 14],
+      [4, 0, 8, 15, 22, 12, 12, 11, 10],
+      [12, 8, 0, 7, 14, 4, 6, 7, 2],
+      [19, 15, 7, 0, 9, 11, 13, 14, 9],
+      [21, 22, 14, 9, 0, 10, 12, 13, 16],
+      [11, 12, 4, 11, 10, 0, 2, 3, 6],
+      [9, 12, 6, 13, 12, 2, 0, 1, 6],
+      [8, 11, 7, 14, 13, 3, 1, 0, 7],
+      [14, 10, 2, 9, 16, 6, 6, 7, 0]
+    ]
+    expect(johnson(graph)).toStrictEqual(expected);
+  });
+
+  it("should return the correct value for graph with negative weights", () => {
+    let graph = init_graph(4);
+    graph[0].push([1, -5]);
+    graph[0].push([2, 2]);
+    graph[0].push([3, 3]);
+    graph[1].push([2, 4]);
+    graph[2].push([3, 1]);
+
+    let expected = [
+      [ 0, -5, -1, 0 ],
+      [ Infinity, 0, 4, 5 ],
+      [ Infinity, Infinity, 0, 1 ],
+      [ Infinity, Infinity, Infinity, 0 ]
+    ]
+    expect(johnson(graph)).toStrictEqual(expected);
+  });
+
+  it("should return the undefined for two node graph with negative-weight cycle", () => {
+    let graph = init_graph(2);
+    add_edge(graph, 0, 1, -1);
+    expect(johnson(graph)).toStrictEqual(undefined);
+  });
+
+  it("should return the undefined for three node graph with negative-weight cycle", () => {
+    let graph = init_graph(3);
+    graph[0].push([1, -1]);
+    graph[0].push([2, 7]);
+    graph[1].push([2, -5]);
+    graph[2].push([0, 4]);
+    expect(johnson(graph)).toStrictEqual(undefined);
+  });
+
+  it("should return the correct value for zero element graph", () => {
+    expect(johnson([])).toStrictEqual([]);
+  });
+
+  it("should return the correct value for single element graph", () => {
+    expect(johnson([[]])).toStrictEqual([[0]]);
+  });
+
+  it("should return the correct value for a linear graph", () => {
+    let linear_graph = init_graph(4);
+    add_edge(linear_graph, 0, 1, 1);
+    add_edge(linear_graph, 1, 2, 2);
+    add_edge(linear_graph, 2, 3, 3);
+
+    let expected = [[0, 1, 3, 6 ], [1, 0, 2, 5], [3, 2, 0, 3], [6, 5, 3, 0]];
+    expect(johnson(linear_graph)).toStrictEqual(expected);
+  });
+
+  it("should return the correct value for a linear graph with unreachable node", () => {
+    let linear_graph = init_graph(3);
+    add_edge(linear_graph, 0, 1, 1);
+
+    let expected = [[0, 1, Infinity], [1, 0, Infinity], [Infinity, Infinity, 0]];
+    expect(johnson(linear_graph)).toStrictEqual(expected);
+  });
+})
+