steve/containers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

changed peek to include an offset

Browse Source
This commit is contained in:
Steve Dudenhoeffer 2022-09-23 22:03:13 -04:00
parent e598bfcfa1
commit 92777666a9
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
priorityqueue.go
View File

@ -58,16 +58,17 @@ func (q *PriorityQueue[T]) Len() int {
return len(q.items) return len(q.items)
} }
func (q *PriorityQueue[T]) Peek() (T, bool) { // offset should be between 0 and Len() - 1
func (q *PriorityQueue[T]) Peek(offset int) (T, bool) {
q.lock.RLock() q.lock.RLock()
defer q.lock.RUnlock() defer q.lock.RUnlock()
if len(q.items) == 0 { if offset < 0 || offset >= len(q.items) {
var def T var t T
return def, false return t, false
} }
return q.items[0].value, true return q.items[offset].value, true
} }
func (q *PriorityQueue[T]) Clear() { func (q *PriorityQueue[T]) Clear() {