fix updates for min index bookkeeping when linkage is not "single"

jdfekete · Aug 15, 2023 · 94afbbc · 94afbbc
1 parent 4138746
commit 94afbbc
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/hcluster.js b/src/hcluster.js
@@ -134,8 +134,16 @@ export function hcluster() {
 
       // update dmin and replace ones that previous pointed to c2 to point to c1
       for (let j = 0; j < n; j++) {
-        if (dMin[j] == c2) {
-          dMin[j] = c1;
+        if (linkage === 'single') {
+          if (dMin[j] === c2) {
+            dMin[j] = c1;
+          }
+        } else if (dMin[j] === c2 || dMin[j] === c1) {
+          for (let k = 0; k < n; k++) {
+            if (distMatrix[j][k] < distMatrix[j][dMin[j]]) {
+              dMin[j] = k;
+            }
+          }
         }
         if (distMatrix[c1][j] < distMatrix[c1][dMin[c1]]) {
           dMin[c1] = j;