mirror of
				https://github.com/krahets/hello-algo.git
				synced 2025-11-04 14:18:20 +08:00 
			
		
		
		
	feat(go/deque): support go array deque (#414)
This commit is contained in:
		
							
								
								
									
										115
									
								
								codes/go/chapter_stack_and_queue/array_deque.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										115
									
								
								codes/go/chapter_stack_and_queue/array_deque.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,115 @@
 | 
				
			|||||||
 | 
					// File: array_deque.go
 | 
				
			||||||
 | 
					// Created Time: 2023-03-13
 | 
				
			||||||
 | 
					// Author: Reanon (793584285@qq.com)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					package chapter_stack_and_queue
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import "fmt"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 基于环形数组实现的双向队列 */
 | 
				
			||||||
 | 
					type arrayDeque struct {
 | 
				
			||||||
 | 
						nums        []int // 用于存储双向队列元素的数组
 | 
				
			||||||
 | 
						front       int   // 队首指针,指向队首元素
 | 
				
			||||||
 | 
						queSize     int   // 双向队列长度
 | 
				
			||||||
 | 
						queCapacity int   // 队列容量(即最大容纳元素数量)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 初始化队列 */
 | 
				
			||||||
 | 
					func newArrayDeque(queCapacity int) *arrayDeque {
 | 
				
			||||||
 | 
						return &arrayDeque{
 | 
				
			||||||
 | 
							nums:        make([]int, queCapacity),
 | 
				
			||||||
 | 
							queCapacity: queCapacity,
 | 
				
			||||||
 | 
							front:       0,
 | 
				
			||||||
 | 
							queSize:     0,
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 获取双向队列的长度 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) size() int {
 | 
				
			||||||
 | 
						return q.queSize
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 判断双向队列是否为空 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) isEmpty() bool {
 | 
				
			||||||
 | 
						return q.queSize == 0
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 计算环形数组索引 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) index(i int) int {
 | 
				
			||||||
 | 
						// 通过取余操作实现数组首尾相连
 | 
				
			||||||
 | 
						// 当 i 越过数组尾部后,回到头部
 | 
				
			||||||
 | 
						// 当 i 越过数组头部后,回到尾部
 | 
				
			||||||
 | 
						return (i + q.queCapacity) % q.queCapacity
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 队首入队 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) pushFirst(num int) {
 | 
				
			||||||
 | 
						if q.queSize == q.queCapacity {
 | 
				
			||||||
 | 
							fmt.Println("双向队列已满")
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						// 队首指针向左移动一位
 | 
				
			||||||
 | 
						// 通过取余操作,实现 front 越过数组头部后回到尾部
 | 
				
			||||||
 | 
						q.front = q.index(q.front - 1)
 | 
				
			||||||
 | 
						// 将 num 添加至队首
 | 
				
			||||||
 | 
						q.nums[q.front] = num
 | 
				
			||||||
 | 
						q.queSize++
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 队尾入队 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) pushLast(num int) {
 | 
				
			||||||
 | 
						if q.queSize == q.queCapacity {
 | 
				
			||||||
 | 
							fmt.Println("双向队列已满")
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						// 计算尾指针,指向队尾索引 + 1
 | 
				
			||||||
 | 
						rear := q.index(q.front + q.queSize)
 | 
				
			||||||
 | 
						// 将 num 添加至队首
 | 
				
			||||||
 | 
						q.nums[rear] = num
 | 
				
			||||||
 | 
						q.queSize++
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 队首出队 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) pollFirst() any {
 | 
				
			||||||
 | 
						num := q.peekFirst()
 | 
				
			||||||
 | 
						// 队首指针向后移动一位
 | 
				
			||||||
 | 
						q.front = q.index(q.front + 1)
 | 
				
			||||||
 | 
						q.queSize--
 | 
				
			||||||
 | 
						return num
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 队尾出队 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) pollLast() any {
 | 
				
			||||||
 | 
						num := q.peekLast()
 | 
				
			||||||
 | 
						q.queSize--
 | 
				
			||||||
 | 
						return num
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 访问队首元素 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) peekFirst() any {
 | 
				
			||||||
 | 
						if q.isEmpty() {
 | 
				
			||||||
 | 
							return nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return q.nums[q.front]
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 访问队尾元素 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) peekLast() any {
 | 
				
			||||||
 | 
						if q.isEmpty() {
 | 
				
			||||||
 | 
							return nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						// 计算尾元素索引
 | 
				
			||||||
 | 
						last := q.index(q.front + q.queSize - 1)
 | 
				
			||||||
 | 
						return q.nums[last]
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/* 获取 Slice 用于打印 */
 | 
				
			||||||
 | 
					func (q *arrayDeque) toSlice() []int {
 | 
				
			||||||
 | 
						// 仅转换有效长度范围内的列表元素
 | 
				
			||||||
 | 
						res := make([]int, q.queSize)
 | 
				
			||||||
 | 
						for i, j := 0, q.front; i < q.queSize; i++ {
 | 
				
			||||||
 | 
							res[i] = q.nums[q.index(j)]
 | 
				
			||||||
 | 
							j++
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return res
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@ -49,6 +49,49 @@ func TestDeque(t *testing.T) {
 | 
				
			|||||||
	fmt.Println("双向队列是否为空 =", isEmpty)
 | 
						fmt.Println("双向队列是否为空 =", isEmpty)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestArrayDeque(t *testing.T) {
 | 
				
			||||||
 | 
						/* 初始化双向队列 */
 | 
				
			||||||
 | 
						// 在 Go 中,将 list 作为双向队列使用
 | 
				
			||||||
 | 
						deque := newArrayDeque(16)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						/* 元素入队 */
 | 
				
			||||||
 | 
						deque.pushLast(3)
 | 
				
			||||||
 | 
						deque.pushLast(2)
 | 
				
			||||||
 | 
						deque.pushLast(5)
 | 
				
			||||||
 | 
						fmt.Print("双向队列 deque = ")
 | 
				
			||||||
 | 
						PrintSlice(deque.toSlice())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						/* 访问元素 */
 | 
				
			||||||
 | 
						peekFirst := deque.peekFirst()
 | 
				
			||||||
 | 
						fmt.Println("队首元素 peekFirst =", peekFirst)
 | 
				
			||||||
 | 
						peekLast := deque.peekLast()
 | 
				
			||||||
 | 
						fmt.Println("队尾元素 peekLast =", peekLast)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						/* 元素入队 */
 | 
				
			||||||
 | 
						deque.pushLast(4)
 | 
				
			||||||
 | 
						fmt.Print("元素 4 队尾入队后 deque = ")
 | 
				
			||||||
 | 
						PrintSlice(deque.toSlice())
 | 
				
			||||||
 | 
						deque.pushFirst(1)
 | 
				
			||||||
 | 
						fmt.Print("元素 1 队首入队后 deque = ")
 | 
				
			||||||
 | 
						PrintSlice(deque.toSlice())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						/* 元素出队 */
 | 
				
			||||||
 | 
						pollFirst := deque.pollFirst()
 | 
				
			||||||
 | 
						fmt.Print("队首出队元素 pollFirst = ", pollFirst, ",队首出队后 deque = ")
 | 
				
			||||||
 | 
						PrintSlice(deque.toSlice())
 | 
				
			||||||
 | 
						pollLast := deque.pollLast()
 | 
				
			||||||
 | 
						fmt.Print("队尾出队元素 pollLast = ", pollLast, ",队尾出队后 deque = ")
 | 
				
			||||||
 | 
						PrintSlice(deque.toSlice())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						/* 获取双向队列的长度 */
 | 
				
			||||||
 | 
						size := deque.size()
 | 
				
			||||||
 | 
						fmt.Println("双向队列长度 size =", size)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						/* 判断双向队列是否为空 */
 | 
				
			||||||
 | 
						isEmpty := deque.isEmpty()
 | 
				
			||||||
 | 
						fmt.Println("双向队列是否为空 =", isEmpty)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func TestLinkedListDeque(t *testing.T) {
 | 
					func TestLinkedListDeque(t *testing.T) {
 | 
				
			||||||
	// 初始化队列
 | 
						// 初始化队列
 | 
				
			||||||
	deque := newLinkedListDeque()
 | 
						deque := newLinkedListDeque()
 | 
				
			||||||
@ -85,7 +128,7 @@ func TestLinkedListDeque(t *testing.T) {
 | 
				
			|||||||
	fmt.Println("队是否为空 =", isEmpty)
 | 
						fmt.Println("队是否为空 =", isEmpty)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// BenchmarkArrayQueue 67.92 ns/op in Mac M1 Pro
 | 
					// BenchmarkLinkedListDeque 67.92 ns/op in Mac M1 Pro
 | 
				
			||||||
func BenchmarkLinkedListDeque(b *testing.B) {
 | 
					func BenchmarkLinkedListDeque(b *testing.B) {
 | 
				
			||||||
	deque := newLinkedListDeque()
 | 
						deque := newLinkedListDeque()
 | 
				
			||||||
	// use b.N for looping
 | 
						// use b.N for looping
 | 
				
			||||||
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user