PQ: min pq added

data-structures/priority-queue/priority_queue.go

@@ -1,7 +1,6 @@
 package pq
 
 import (
-	"fmt"
 	"github.com/arnauddri/algorithms/data-structures/heap"
 	"github.com/arnauddri/algorithms/data-structures/queue"
 )
@@ -26,12 +25,18 @@ type PQ struct {
 	data heap.Heap
 }
 
-func New() (q *PQ) {
+func NewMax() (q *PQ) {
 	return &PQ{
 		data: *heap.NewMax(),
 	}
 }
 
+func NewMin() (q *PQ) {
+	return &PQ{
+		data: *heap.NewMin(),
+	}
+}
+
 func (pq *PQ) Len() int {
 	return pq.data.Len()
 }
@@ -50,7 +55,6 @@ func (pq *PQ) ChangePriority(val interface{}, priority int) {
 	popped := pq.Extract()
 
 	for val != popped.Value {
-		fmt.Println(val, popped.Value)
 		if pq.Len() == 0 {
 			panic("Item not found")
 		}

data-structures/priority-queue/priority_queue_test.go

@@ -5,8 +5,8 @@ import (
 	"testing"
 )
 
-func TestPriorityQueue(t *testing.T) {
-	h := New()
+func TestMaxPriorityQueue(t *testing.T) {
+	h := NewMax()
 
 	h.Insert(*NewItem(8, 10))
 	h.Insert(*NewItem(7, 11))
@@ -32,8 +32,35 @@ func TestPriorityQueue(t *testing.T) {
 	}
 }
 
+func TestMinPriorityQueue(t *testing.T) {
+	h := NewMin()
+
+	h.Insert(*NewItem(8, 10))
+	h.Insert(*NewItem(7, 11))
+	h.Insert(*NewItem(6, 12))
+	h.Insert(*NewItem(3, 13))
+	h.Insert(*NewItem(1, 14))
+	h.Insert(*NewItem(0, 15))
+	h.Insert(*NewItem(2, 16))
+	h.Insert(*NewItem(4, 17))
+	h.Insert(*NewItem(9, 18))
+	h.Insert(*NewItem(5, 19))
+
+	sorted := make([]Item, 0)
+	for h.Len() > 0 {
+		sorted = append(sorted, h.Extract())
+	}
+
+	for i := 0; i < len(sorted)-2; i++ {
+		if sorted[i].Priority > sorted[i+1].Priority {
+			fmt.Println(sorted)
+			t.Error()
+		}
+	}
+}
+
 func TestChangePriority(t *testing.T) {
-	h := New()
+	h := NewMax()
 
 	h.Insert(*NewItem(8, 10))
 	h.Insert(*NewItem(7, 11))