49 lines
676 B
Go
49 lines
676 B
Go
|
package stack
|
||
|
|
||
|
import "fmt"
|
||
|
|
||
|
type Stack[T any] struct {
|
||
|
arr []T
|
||
|
size int
|
||
|
}
|
||
|
|
||
|
func NewStack[T any]() *Stack[T] {
|
||
|
return &Stack[T]{
|
||
|
arr: make([]T, 0),
|
||
|
size: 0,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (s *Stack[T]) Clear() {
|
||
|
s.arr = make([]T, 0)
|
||
|
s.size = 0
|
||
|
}
|
||
|
|
||
|
func (s *Stack[T]) Size() int {
|
||
|
return s.size
|
||
|
}
|
||
|
|
||
|
func (s *Stack[T]) Pop() (*T, error) {
|
||
|
if s.size == 0 {
|
||
|
return nil, fmt.Errorf("tried to pop empty stack")
|
||
|
}
|
||
|
t := s.arr[s.size-1]
|
||
|
s.arr = s.arr[:s.size-1]
|
||
|
s.size--
|
||
|
return &t, nil
|
||
|
}
|
||
|
|
||
|
func (s *Stack[T]) Push(t T) {
|
||
|
s.arr = append(s.arr, t)
|
||
|
s.size++
|
||
|
}
|
||
|
|
||
|
func (s *Stack[T]) Peek() T {
|
||
|
t := s.arr[s.size-1]
|
||
|
return t
|
||
|
}
|
||
|
|
||
|
func (s *Stack[T]) PeekPtr() *T {
|
||
|
return &(s.arr[s.size-1])
|
||
|
}
|