zishuo/lancet
1
0
Fork 0
mirror of https://github.com/duke-git/lancet.git synced 2025-10-05 07:26:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: add unit test for PriorityQueue Dequeue function

Browse Source
This commit is contained in:
dudaodong
2022-04-26 10:56:24 +08:00
parent 955de2bdbf
commit 652916b7d7
2 changed files with 23 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
datastructure/queue/priorityqueue.go
View File

@@ -76,7 +76,7 @@ func (q *PriorityQueue[T]) Dequeue() (T, bool) {
// swim when child's key is larger than parent's key, exchange them.
func (q *PriorityQueue[T]) swim(index int) {
for index > 1 && q.comparator.Compare(index/2, index) < 0 {
for index > 1 && q.comparator.Compare(q.items[index/2], q.items[index]) < 0 {
q.swap(index, index/2)
index = index / 2
}
@@ -89,11 +89,11 @@ func (q *PriorityQueue[T]) sink(index int) {
j := 2 * index
// get larger child node index
if j < q.size && q.comparator.Compare(j, j+1) < 0 {
if j < q.size && q.comparator.Compare(q.items[j], q.items[j+1]) < 0 {
j++
}
// if parent larger than child, stop
if !(q.comparator.Compare(index, j) < 0) {
if !(q.comparator.Compare(q.items[index], q.items[j]) < 0) {
break
}