优化代码

util/stack/stack.go

@@ -0,0 +1,41 @@
+package stack
+
+// Stack 是一个通用类型的栈结构
+type Stack[T any] struct {
+	items []*T
+}
+
+// Push 将元素压入栈顶
+func (s *Stack[T]) Push(item *T) {
+	s.items = append(s.items, item)
+}
+
+// Pop 弹出并返回栈顶元素，如果栈为空则返回错误
+func (s *Stack[T]) Pop() *T {
+	if len(s.items) == 0 {
+		return nil
+	}
+
+	index := len(s.items) - 1
+	item := s.items[index]
+	s.items = s.items[:index]
+	return item
+}
+
+// Peek 返回栈顶元素但不移除，如果栈为空则返回错误
+func (s *Stack[T]) Peek() *T {
+	if len(s.items) == 0 {
+		return nil
+	}
+	return s.items[len(s.items)-1]
+}
+
+// IsEmpty 检查栈是否为空
+func (s *Stack[T]) IsEmpty() bool {
+	return len(s.items) == 0
+}
+
+// Size 返回栈中元素的数量
+func (s *Stack[T]) Size() int {
+	return len(s.items)
+}