aboutsummaryrefslogtreecommitdiff
path: root/internal/priorityqueue/sift_down.go
diff options
context:
space:
mode:
authorGravatar Runxi Yu2026-03-29 14:14:21 +0000
committerGravatar Runxi Yu2026-03-29 14:14:21 +0000
commit69452f58fb0d6c1ed561df1e25efe51e5b3089fd (patch)
tree3e9cf39ffc197acca3da05bc4162581767ac2b64 /internal/priorityqueue/sift_down.go
parentcommitquery: Use internal/heap for the priority queue (diff)
signatureNo signature
internal/priorityqueue: Actually just make our own priority queue
Diffstat (limited to 'internal/priorityqueue/sift_down.go')
-rw-r--r--internal/priorityqueue/sift_down.go23
1 files changed, 23 insertions, 0 deletions
diff --git a/internal/priorityqueue/sift_down.go b/internal/priorityqueue/sift_down.go
new file mode 100644
index 00000000..83e3fd40
--- /dev/null
+++ b/internal/priorityqueue/sift_down.go
@@ -0,0 +1,23 @@
+package priorityqueue
+
+func (queue *Queue[T]) siftDown(idx int) {
+ for {
+ left := idx*2 + 1
+ if left >= len(queue.items) {
+ return
+ }
+
+ best := left
+ right := left + 1
+ if right < len(queue.items) && queue.less(queue.items[right], queue.items[left]) {
+ best = right
+ }
+
+ if !queue.less(queue.items[best], queue.items[idx]) {
+ return
+ }
+
+ queue.items[idx], queue.items[best] = queue.items[best], queue.items[idx]
+ idx = best
+ }
+}