hello-algo/codes/go/chapter_stack_and_queue/array_queue.go

79 lines
1.7 KiB
Go
Raw Normal View History

// File: array_queue.go
// Created Time: 2022-11-28
// Author: Reanon (793584285@qq.com)
package chapter_stack_and_queue
/* 基于环形数组实现的队列 */
type arrayQueue struct {
2023-02-01 03:23:29 +08:00
nums []int // 用于存储队列元素的数组
front int // 队首指针,指向队首元素
queSize int // 队列长度
queCapacity int // 队列容量(即最大容纳元素数量)
}
/* 初始化队列 */
2023-02-01 03:23:29 +08:00
func newArrayQueue(queCapacity int) *arrayQueue {
return &arrayQueue{
2023-02-01 03:23:29 +08:00
nums: make([]int, queCapacity),
queCapacity: queCapacity,
front: 0,
queSize: 0,
}
}
/* 获取队列的长度 */
func (q *arrayQueue) size() int {
2023-02-01 03:23:29 +08:00
return q.queSize
}
/* 判断队列是否为空 */
func (q *arrayQueue) isEmpty() bool {
2023-02-01 03:23:29 +08:00
return q.queSize == 0
}
/* 入队 */
func (q *arrayQueue) push(num int) {
2023-02-01 03:23:29 +08:00
// 当 rear == queCapacity 表示队列已满
if q.queSize == q.queCapacity {
return
}
// 计算队尾指针,指向队尾索引 + 1
// 通过取余操作实现 rear 越过数组尾部后回到头部
2023-02-01 03:23:29 +08:00
rear := (q.front + q.queSize) % q.queCapacity
// 将 num 添加至队尾
2023-02-01 03:23:29 +08:00
q.nums[rear] = num
q.queSize++
}
/* 出队 */
func (q *arrayQueue) pop() any {
2023-02-01 03:23:29 +08:00
num := q.peek()
if num == nil {
return nil
}
// 队首指针向后移动一位,若越过尾部,则返回到数组头部
2023-02-01 03:23:29 +08:00
q.front = (q.front + 1) % q.queCapacity
q.queSize--
return num
}
/* 访问队首元素 */
func (q *arrayQueue) peek() any {
if q.isEmpty() {
return nil
}
2023-02-01 03:23:29 +08:00
return q.nums[q.front]
}
/* 获取 Slice 用于打印 */
func (q *arrayQueue) toSlice() []int {
2023-02-01 03:23:29 +08:00
rear := (q.front + q.queSize)
if rear >= q.queCapacity {
rear %= q.queCapacity
return append(q.nums[q.front:], q.nums[:rear]...)
}
return q.nums[q.front:rear]
}