Kruskal adds edges in ascending weight order whenever they join different components..
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 | "use strict"; // Kruskal Minimum Spanning Tree in JavaScript // // PURPOSE // Kruskal adds edges in ascending weight order whenever they join different components. // // METHOD // Sort edge indices by weight using a stable bottom-up merge. Use union by size and path // halving to accept edges joining different roots. Stop after n-1 accepted edges. // // COMPLEXITY // O(E log E + V) time and O(V+E) auxiliary space. // // CONTRACT // n>=1, parallel edge arrays have m entries and valid vertex IDs. Weights nonnegative; total // fits signed 32-bit. Returns -1 for a disconnected graph. // Public APIs infer array lengths except in C, which accepts pointer/length pairs. Vertex // counts and matrix dimensions remain explicit. Arrays must contain signed 32-bit integers. // Sorted order, correct tree topology, and caller-owned data-structure invariants are // preconditions; checking them on every query would change query complexity. // Scalar and shape checks are included. Value-range checks are included where needed to // prevent invalid indexing or nontermination. Allocation sizes are capped at 4,000,000 // workspace entries. Arithmetic outside Python raises an error on 32-bit overflow; Python // uses arbitrary precision. Native allocation failures propagate. In-place outputs may be // partly updated after arithmetic failure. These are single-threaded, in-memory // implementations. // // LANGUAGE DETAILS // Node.js 18+: Int32Array inputs and workspaces provide compact integer storage. Automatic // lengths and checked arithmetic reject values outside signed 32-bit. Invalid inputs or // overflow throw RangeError. // // EXAMPLE // Expected return: 6. // Run: node example.js // Tests for this exact source artifact run before publication. The accompanying source // archive includes unit cases, runtime results, and source hashes. function alg_require(valid) { if (!valid) throw new RangeError("invalid algorithm input"); } function alg_checked(v) { if (!Number.isInteger(v) || v < -2147483648 || v > 2147483647) throw new RangeError("32-bit result overflow"); return v; } function alg_add(a,b) { return alg_checked(a+b); } function alg_sub(a,b) { return alg_checked(a-b); } function alg_mul(a,b) { return alg_checked(a*b); } function alg_size(n) { alg_require(Number.isInteger(n) && n>=0 && n<=4000000); return n; } function kruskal(from_vertex, to_vertex, weight, n) { alg_require(from_vertex instanceof Int32Array); alg_require(to_vertex instanceof Int32Array); alg_require(weight instanceof Int32Array); alg_require(Number.isInteger(n) && n >= -2147483648 && n <= 2147483647); let m = from_vertex.length; alg_require(((n >= 0) && (n <= 1000000))); alg_require(((m >= 0) && (m <= 1000000))); alg_require((from_vertex.length == m)); alg_require((to_vertex.length == m)); alg_require((weight.length == m)); alg_require((n >= 1)); for (let alg_i = 0; alg_i < from_vertex.length; alg_i++) { let alg_value = from_vertex[alg_i]; alg_require(((alg_value >= 0) && (alg_value < n))); } for (let alg_i = 0; alg_i < to_vertex.length; alg_i++) { let alg_value = to_vertex[alg_i]; alg_require(((alg_value >= 0) && (alg_value < n))); } let order = new Int32Array(alg_size(alg_add(m, 1))); let buffer = new Int32Array(alg_size(alg_add(m, 1))); let parent = new Int32Array(alg_size(n)); let size = new Int32Array(alg_size(n)); let i = 0; let j = 0; let k = 0; let width = 1; let left = 0; let mid = 0; let right = 0; let edge = 0; let u = 0; let v = 0; let temp = 0; let accepted = 0; let total = 0; while ((i < m)) { order[i] = i; i = alg_add(i, 1); } while ((width < m)) { left = 0; while ((left < m)) { mid = Math.min(alg_add(left, width), m); right = Math.min(alg_add(left, alg_mul(2, width)), m); i = left; j = mid; k = left; while ((k < right)) { if (((i < mid) && ((j >= right) || (weight[order[i]] <= weight[order[j]])))) { buffer[k] = order[i]; i = alg_add(i, 1); } else { buffer[k] = order[j]; j = alg_add(j, 1); } k = alg_add(k, 1); } left = alg_add(left, alg_mul(2, width)); } i = 0; while ((i < m)) { order[i] = buffer[i]; i = alg_add(i, 1); } width = alg_mul(width, 2); } i = 0; while ((i < n)) { parent[i] = i; size[i] = 1; i = alg_add(i, 1); } while (((edge < m) && (accepted < alg_sub(n, 1)))) { u = from_vertex[order[edge]]; v = to_vertex[order[edge]]; while ((parent[u] != u)) { parent[u] = parent[parent[u]]; u = parent[u]; } while ((parent[v] != v)) { parent[v] = parent[parent[v]]; v = parent[v]; } if ((u != v)) { if ((size[u] < size[v])) { temp = u; u = v; v = temp; } parent[v] = u; size[u] = alg_add(size[u], size[v]); accepted = alg_add(accepted, 1); total = alg_add(total, weight[order[edge]]); } edge = alg_add(edge, 1); } if ((accepted != alg_sub(n, 1))) { return alg_sub(0, 1); } return total; } let from_vertex = new Int32Array([0, 1, 0]); let to_vertex = new Int32Array([1, 2, 2]); let weight = new Int32Array([4, 2, 10]); let n = 3; let result = kruskal(from_vertex, to_vertex, weight, n); console.log(result); |