From c8dcf8993fa32e442777dd3aaf0116518a68a71a Mon Sep 17 00:00:00 2001
From: Yuriy Glukhov <yglukhov@users.noreply.github.com>
Date: Fri, 27 Jan 2017 09:22:17 +0200
Subject: [PATCH] Added heapqueue.del (#5289)

---
 lib/pure/collections/heapqueue.nim | 46 +++++++++++++++++++++++-------
 1 file changed, 35 insertions(+), 11 deletions(-)

diff --git a/lib/pure/collections/heapqueue.nim b/lib/pure/collections/heapqueue.nim
index 149a1c9fc2..abe20e5568 100644
--- a/lib/pure/collections/heapqueue.nim
+++ b/lib/pure/collections/heapqueue.nim
@@ -58,7 +58,7 @@ proc siftup[T](heap: var HeapQueue[T], p: int) =
   # to its final resting place (by sifting its parents down).
   heap[pos] = newitem
   siftdown(heap, startpos, pos)
-    
+
 proc push*[T](heap: var HeapQueue[T], item: T) =
   ## Push item onto heap, maintaining the heap invariant.
   (seq[T](heap)).add(item)
@@ -74,6 +74,12 @@ proc pop*[T](heap: var HeapQueue[T]): T =
   else:
     result = lastelt
 
+proc del*[T](heap: var HeapQueue[T], index: int) =
+  ## Removes element at `index`, maintaining the heap invariant.
+  swap(seq[T](heap)[^1], seq[T](heap)[index])
+  seq[T](heap).setLen(heap.len - 1)
+  heap.siftup(index)
+
 proc replace*[T](heap: var HeapQueue[T], item: T): T =
   ## Pop and return the current smallest value, and add the new item.
   ## This is more efficient than pop() followed by push(), and can be
@@ -95,13 +101,31 @@ proc pushpop*[T](heap: var HeapQueue[T], item: T): T =
   return item
 
 when isMainModule:
-  # Simple sanity test
-  var heap = newHeapQueue[int]()
-  let data = [1, 3, 5, 7, 9, 2, 4, 6, 8, 0]
-  for item in data:
-    push(heap, item)
-  doAssert(heap[0] == 0)
-  var sort = newSeq[int]()
-  while heap.len > 0:
-    sort.add(pop(heap))
-  doAssert(sort == @[0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
+  block: # Simple sanity test
+    var heap = newHeapQueue[int]()
+    let data = [1, 3, 5, 7, 9, 2, 4, 6, 8, 0]
+    for item in data:
+      push(heap, item)
+    doAssert(heap[0] == 0)
+    var sort = newSeq[int]()
+    while heap.len > 0:
+      sort.add(pop(heap))
+    doAssert(sort == @[0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
+
+  block: # Test del
+    var heap = newHeapQueue[int]()
+    let data = [1, 3, 5, 7, 9, 2, 4, 6, 8, 0]
+    for item in data: push(heap, item)
+
+    heap.del(0)
+    doAssert(heap[0] == 1)
+
+    heap.del(seq[int](heap).find(7))
+    heap.del(seq[int](heap).find(5))
+    heap.del(seq[int](heap).find(6))
+    heap.del(seq[int](heap).find(2))
+
+    var sort = newSeq[int]()
+    while heap.len > 0:
+      sort.add(pop(heap))
+    doAssert(sort == @[1, 3, 4, 8, 9])