83 lines
1.1 KiB
Go
83 lines
1.1 KiB
Go
|
package coll
|
||
|
|
|
||
|
|
// Queue 队列, 先进先出
|
||
|
|
type Queue[T any] []T
|
||
|
|
|
||
|
|
func NewQueue[T any](elems ...T) Queue[T] {
|
||
|
|
if len(elems) > 0 {
|
||
|
|
data := make([]T, len(elems))
|
||
|
|
copy(data, elems)
|
||
|
|
return data
|
||
|
|
} else {
|
||
|
|
return Queue[T]{}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Push 尾部插入元素
|
||
|
|
func (q *Queue[T]) Push(elem T) {
|
||
|
|
if q == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
*q = append(*q, elem)
|
||
|
|
}
|
||
|
|
|
||
|
|
// PushN 尾部插入多个元素
|
||
|
|
func (q *Queue[T]) PushN(elems ...T) {
|
||
|
|
if q == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if len(elems) <= 0 {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
*q = append(*q, elems...)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Pop 移除并返回头部元素
|
||
|
|
func (q *Queue[T]) Pop() (T, bool) {
|
||
|
|
var elem T
|
||
|
|
if q == nil || len(*q) <= 0 {
|
||
|
|
return elem, false
|
||
|
|
}
|
||
|
|
elem = (*q)[0]
|
||
|
|
*q = (*q)[1:]
|
||
|
|
return elem, true
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q *Queue[T]) PopN(n int) []T {
|
||
|
|
if q == nil {
|
||
|
|
return []T{}
|
||
|
|
}
|
||
|
|
|
||
|
|
var popElems []T
|
||
|
|
if n <= 0 {
|
||
|
|
return []T{}
|
||
|
|
}
|
||
|
|
|
||
|
|
l := len(*q)
|
||
|
|
if n >= l {
|
||
|
|
popElems = *q
|
||
|
|
*q = []T{}
|
||
|
|
return *q
|
||
|
|
}
|
||
|
|
|
||
|
|
popElems = (*q)[:n]
|
||
|
|
*q = (*q)[n:]
|
||
|
|
return popElems
|
||
|
|
}
|
||
|
|
|
||
|
|
// Clear 移除所有元素
|
||
|
|
func (q *Queue[T]) Clear() {
|
||
|
|
if q == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
*q = []T{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q Queue[T]) IsEmpty() bool {
|
||
|
|
return len(q) <= 0
|
||
|
|
}
|
||
|
|
|
||
|
|
func (q Queue[T]) Size() int {
|
||
|
|
return len(q)
|
||
|
|
}
|