package stack

import (
	"fmt"
	"strings"
)

// 使用数组实现的栈
type ArrayStack[T any] struct {
	values []T
}

func NewArrayStack[T any]() *ArrayStack[T] {
	return &ArrayStack[T]{values: make([]T, 0, 10)}
}

func (a *ArrayStack[T]) String() string {
	var ans strings.Builder
	ans.WriteString("[")
	for i, value := range a.values {
		if i+1 == len(a.values) {
			ans.WriteString(fmt.Sprintf("%v", value))
			break
		}
		ans.WriteString(fmt.Sprintf("%v, ", value))
	}
	ans.WriteString("]")
	return ans.String()
}

func (a *ArrayStack[T]) IsEmpty() bool {
	return len(a.values) == 0
}

func (a *ArrayStack[T]) Size() int {
	return len(a.values)
}

func (a *ArrayStack[T]) Push(e T) {
	a.values = append(a.values, e)
}

func (a *ArrayStack[T]) Pop() T {
	i := len(a.values) - 1
	ans := a.values[i]
	a.values = a.values[:i]
	return ans
}

func (a *ArrayStack[T]) Peek() T {
	return a.values[len(a.values)-1]
}