You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
571 B
Go
34 lines
571 B
Go
// Copied from https://gist.github.com/hedhyw/d52bfdc27befe56ffc59b948086fcd9e
|
|
|
|
package stack
|
|
|
|
type Stack[T any] struct {
|
|
elements []T
|
|
}
|
|
|
|
func NewStack[T any](capacity int) *Stack[T] {
|
|
return &Stack[T]{
|
|
elements: make([]T, 0, capacity),
|
|
}
|
|
}
|
|
|
|
func (s *Stack[T]) Push(el T) {
|
|
s.elements = append(s.elements, el)
|
|
}
|
|
|
|
func (s *Stack[T]) Len() int {
|
|
return len(s.elements)
|
|
}
|
|
|
|
func (s *Stack[T]) Pop() (el T, ok bool) {
|
|
if len(s.elements) == 0 {
|
|
return el, false
|
|
}
|
|
|
|
end := len(s.elements) - 1
|
|
el = s.elements[end]
|
|
s.elements = s.elements[:end]
|
|
|
|
return el, true
|
|
}
|