容器

collections/queue.go

@@ -0,0 +1,71 @@
+package collections
+
+import "sync"
+
+// 数组队列，先进先出
+type ArrayQueue struct {
+	array []interface{} // 底层切片
+	size  int           // 队列的元素数量
+	lock  sync.Mutex    // 为了并发安全使用的锁
+}
+
+func NewArrayQueue[T any]() *ArrayQueue {
+	return &ArrayQueue{}
+}
+
+// 入队
+func (queue *ArrayQueue) Add(v interface{}) {
+	queue.lock.Lock()
+	defer queue.lock.Unlock()
+
+	// 放入切片中，后进的元素放在数组最后面
+	queue.array = append(queue.array, v)
+
+	// 队中元素数量+1
+	queue.size = queue.size + 1
+}
+
+// 出队
+func (queue *ArrayQueue) Remove() interface{} {
+	queue.lock.Lock()
+	defer queue.lock.Unlock()
+
+	// 队中元素已空
+	if queue.size == 0 {
+		panic("empty")
+	}
+
+	// 队列最前面元素
+	v := queue.array[0]
+
+	/*    直接原位移动，但缩容后继的空间不会被释放
+	      for i := 1; i < queue.size; i++ {
+	          // 从第一位开始进行数据移动
+	          queue.array[i-1] = queue.array[i]
+	      }
+	      // 原数组缩容
+	      queue.array = queue.array[0 : queue.size-1]
+	*/
+
+	// 创建新的数组，移动次数过多
+	newArray := make([]interface{}, queue.size-1)
+	for i := 1; i < queue.size; i++ {
+		// 从老数组的第一位开始进行数据移动
+		newArray[i-1] = queue.array[i]
+	}
+	queue.array = newArray
+
+	// 队中元素数量-1
+	queue.size = queue.size - 1
+	return v
+}
+
+// 栈大小
+func (queue *ArrayQueue) Size() int {
+	return queue.size
+}
+
+// 栈是否为空
+func (queue *ArrayQueue) IsEmpty() bool {
+	return queue.size == 0
+}