mirror of
https://github.com/krahets/hello-algo.git
synced 2025-12-19 07:17:54 +08:00
Add Go codes to docs, including
the chapter of stack and queue, the chapter of tree.
This commit is contained in:
@@ -4,50 +4,45 @@
|
||||
|
||||
package chapter_stack_and_queue
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// ArrayQueue 基于环形数组实现的队列, 不支持扩容
|
||||
/* 基于环形数组实现的队列 */
|
||||
type ArrayQueue struct {
|
||||
data []any // 用于存储队列元素的数组
|
||||
data []int // 用于存储队列元素的数组
|
||||
capacity int // 队列容量(即最多容量的元素个数)
|
||||
head int // 头指针,指向队首
|
||||
tail int // 尾指针,指向队尾 + 1
|
||||
front int // 头指针,指向队首
|
||||
rear int // 尾指针,指向队尾 + 1
|
||||
}
|
||||
|
||||
// NewArrayQueue 基于环形数组实现的队列
|
||||
func NewArrayQueue(capacity int) *ArrayQueue {
|
||||
return &ArrayQueue{
|
||||
data: make([]any, capacity),
|
||||
data: make([]int, capacity),
|
||||
capacity: capacity,
|
||||
head: 0,
|
||||
tail: 0,
|
||||
front: 0,
|
||||
rear: 0,
|
||||
}
|
||||
}
|
||||
|
||||
// Size 获取队列的长度
|
||||
func (q *ArrayQueue) Size() int {
|
||||
size := (q.capacity + q.tail - q.head) % q.capacity
|
||||
size := (q.capacity + q.rear - q.front) % q.capacity
|
||||
return size
|
||||
}
|
||||
|
||||
// IsEmpty 判断队列是否为空
|
||||
func (q *ArrayQueue) IsEmpty() bool {
|
||||
return q.tail-q.head == 0
|
||||
return q.rear-q.front == 0
|
||||
}
|
||||
|
||||
// Offer 入队
|
||||
func (q *ArrayQueue) Offer(v any) {
|
||||
// 当 tail == capacity 表示队列已满
|
||||
func (q *ArrayQueue) Offer(v int) {
|
||||
// 当 rear == capacity 表示队列已满
|
||||
if q.Size() == q.capacity {
|
||||
return
|
||||
}
|
||||
// 尾结点后添加
|
||||
q.data[q.tail] = v
|
||||
q.data[q.rear] = v
|
||||
// 尾指针向后移动一位,越过尾部后返回到数组头部
|
||||
q.tail = (q.tail + 1) % q.capacity
|
||||
q.rear = (q.rear + 1) % q.capacity
|
||||
}
|
||||
|
||||
// Poll 出队
|
||||
@@ -55,9 +50,9 @@ func (q *ArrayQueue) Poll() any {
|
||||
if q.IsEmpty() {
|
||||
return nil
|
||||
}
|
||||
v := q.data[q.head]
|
||||
v := q.data[q.front]
|
||||
// 队头指针向后移动一位,若越过尾部则返回到数组头部
|
||||
q.head = (q.head + 1) % q.capacity
|
||||
q.front = (q.front + 1) % q.capacity
|
||||
return v
|
||||
}
|
||||
|
||||
@@ -66,27 +61,11 @@ func (q *ArrayQueue) Peek() any {
|
||||
if q.IsEmpty() {
|
||||
return nil
|
||||
}
|
||||
v := q.data[q.head]
|
||||
v := q.data[q.front]
|
||||
return v
|
||||
}
|
||||
|
||||
func (q *ArrayQueue) Print() {
|
||||
fmt.Println(q.toString())
|
||||
}
|
||||
|
||||
// toString 通过字符串的方式输出
|
||||
func (q *ArrayQueue) toString() string {
|
||||
// 为空时
|
||||
if q.IsEmpty() {
|
||||
return "empty items"
|
||||
}
|
||||
var builder strings.Builder
|
||||
size := q.Size()
|
||||
str := fmt.Sprintf("%+v", q.data[q.head])
|
||||
for i := 1; i < size; i++ {
|
||||
builder.WriteString(str + " -> ")
|
||||
str = fmt.Sprintf("%+v", q.data[(i+q.head)%q.capacity])
|
||||
}
|
||||
builder.WriteString(str)
|
||||
return builder.String()
|
||||
// 获取 Slice 用于打印
|
||||
func (s *ArrayQueue) toSlice() []int {
|
||||
return s.data[s.front:s.rear]
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user