mirror of https://github.com/emirpasic/gods
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.6 KiB
Go
87 lines
2.6 KiB
Go
2 years ago
|
// Copyright (c) 2015, Emir Pasic. All rights reserved.
|
||
|
// Use of this source code is governed by a BSD-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
// Package priorityqueue implements a priority queue backed by binary queue.
|
||
|
//
|
||
|
// An unbounded priority queue based on a priority queue.
|
||
|
// The elements of the priority queue are ordered by a comparator provided at queue construction time.
|
||
|
//
|
||
|
// The heap of this queue is the least/smallest element with respect to the specified ordering.
|
||
|
// If multiple elements are tied for least value, the heap is one of those elements arbitrarily.
|
||
|
//
|
||
|
// Structure is not thread safe.
|
||
|
//
|
||
|
// References: https://en.wikipedia.org/wiki/Priority_queue
|
||
|
package priorityqueue
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"github.com/emirpasic/gods/queues"
|
||
|
"github.com/emirpasic/gods/trees/binaryheap"
|
||
|
"github.com/emirpasic/gods/utils"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
// Assert Queue implementation
|
||
|
var _ queues.Queue = (*Queue)(nil)
|
||
|
|
||
|
// Queue holds elements in an array-list
|
||
|
type Queue struct {
|
||
|
heap *binaryheap.Heap
|
||
|
Comparator utils.Comparator
|
||
|
}
|
||
|
|
||
|
// NewWith instantiates a new empty queue with the custom comparator.
|
||
|
func NewWith(comparator utils.Comparator) *Queue {
|
||
|
return &Queue{heap: binaryheap.NewWith(comparator), Comparator: comparator}
|
||
|
}
|
||
|
|
||
|
// Enqueue adds a value to the end of the queue
|
||
|
func (queue *Queue) Enqueue(value interface{}) {
|
||
|
queue.heap.Push(value)
|
||
|
}
|
||
|
|
||
|
// Dequeue removes first element of the queue and returns it, or nil if queue is empty.
|
||
|
// Second return parameter is true, unless the queue was empty and there was nothing to dequeue.
|
||
|
func (queue *Queue) Dequeue() (value interface{}, ok bool) {
|
||
|
return queue.heap.Pop()
|
||
|
}
|
||
|
|
||
|
// Peek returns top element on the queue without removing it, or nil if queue is empty.
|
||
|
// Second return parameter is true, unless the queue was empty and there was nothing to peek.
|
||
|
func (queue *Queue) Peek() (value interface{}, ok bool) {
|
||
|
return queue.heap.Peek()
|
||
|
}
|
||
|
|
||
|
// Empty returns true if queue does not contain any elements.
|
||
|
func (queue *Queue) Empty() bool {
|
||
|
return queue.heap.Empty()
|
||
|
}
|
||
|
|
||
|
// Size returns number of elements within the queue.
|
||
|
func (queue *Queue) Size() int {
|
||
|
return queue.heap.Size()
|
||
|
}
|
||
|
|
||
|
// Clear removes all elements from the queue.
|
||
|
func (queue *Queue) Clear() {
|
||
|
queue.heap.Clear()
|
||
|
}
|
||
|
|
||
|
// Values returns all elements in the queue.
|
||
|
func (queue *Queue) Values() []interface{} {
|
||
|
return queue.heap.Values()
|
||
|
}
|
||
|
|
||
|
// String returns a string representation of container
|
||
|
func (queue *Queue) String() string {
|
||
|
str := "PriorityQueue\n"
|
||
|
values := make([]string, queue.heap.Size(), queue.heap.Size())
|
||
|
for index, value := range queue.heap.Values() {
|
||
|
values[index] = fmt.Sprintf("%v", value)
|
||
|
}
|
||
|
str += strings.Join(values, ", ")
|
||
|
return str
|
||
|
}
|