mirror of
https://github.com/duanhf2012/origin.git
synced 2026-02-04 06:54:45 +08:00
42 lines
801 B
Go
42 lines
801 B
Go
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)
|
|
}
|