Fix bug in ancestor code

huynhtrankhanh · Nov 2, 2024 · 5752015 · 5752015
1 parent 1c1b27a
commit 5752015
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/programs/DisjointSetUnion.js b/programs/DisjointSetUnion.js
@@ -18,8 +18,9 @@ procedure('ancestor', { vertex: int8, work: int8 }, () => {
     if (sLess(retrieve('dsu', coerceInt64(get('work')))[0], coerceInt8(0))) {
       ;('break')
     }
+    set('vertex', retrieve('dsu', coerceInt64(get('work')))[0])
     store('dsu', coerceInt64(get('work')), [retrieve('result', 0)[0]])
-    set('work', retrieve('dsu', coerceInt64(get('work')))[0])
+    set('work', get('vertex'))
   })
 })