// CookieJar - A contestant's algorithm toolbox // Copyright (c) 2013 Peter Szilagyi. All rights reserved. // // CookieJar is dual licensed: you can redistribute it and/or modify it under // the terms of the GNU General Public License as published by the Free Software // Foundation, either version 3 of the License, or (at your option) any later // version. // // The toolbox is distributed in the hope that it will be useful, but WITHOUT // ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or // FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for // more details. // // Alternatively, the CookieJar toolbox may be used in accordance with the terms // and conditions contained in a signed written agreement between you and the // author(s). // Package prque implements a priority queue data structure supporting arbitrary // value types and float priorities. // // The reasoning behind using floats for the priorities vs. ints or interfaces // was larger flexibility without sacrificing too much performance or code // complexity. // // If you would like to use a min-priority queue, simply negate the priorities. // // Internally the queue is based on the standard heap package working on a // sortable version of the block based stack. package prque import ( "container/heap" ) // Priority queue data structure. type Prque struct { cont *sstack } // Creates a new priority queue. func New() *Prque { return &Prque{newSstack()} } // Pushes a value with a given priority into the queue, expanding if necessary. func (p *Prque) Push(data interface{}, priority float32) { heap.Push(p.cont, &item{data, priority}) } // Pops the value with the greates priority off the stack and returns it. // Currently no shrinking is done. func (p *Prque) Pop() (interface{}, float32) { item := heap.Pop(p.cont).(*item) return item.value, item.priority } // Pops only the item from the queue, dropping the associated priority value. func (p *Prque) PopItem() interface{} { return heap.Pop(p.cont).(*item).value } // Checks whether the priority queue is empty. func (p *Prque) Empty() bool { return p.cont.Len() == 0 } // Returns the number of element in the priority queue. func (p *Prque) Size() int { return p.cont.Len() } // Clears the contents of the priority queue. func (p *Prque) Reset() { *p = *New() }