stack.go

package stack

import (
	"fmt"
	"strings"
)

type Stack[T any] struct {
	data []T
	idx  int
}

// New Stack
func New[T any]() *Stack[T] {
	return &Stack[T]{idx: -1}
}

// Top returns top element
func (s *Stack[T]) Top() T {
	return s.data[s.idx]
}

// Push given element `a` into stack
func (s *Stack[T]) Push(a T) {
	s.idx++
	if s.idx < len(s.data) {
		s.data[s.idx] = a
	} else {
		s.data = append(s.data, a)
	}
}

// Pop element from stack,
// return `nil` when stack is empty.
func (s *Stack[T]) Pop() (v T, ok bool) {
	if s.idx > -1 {
		s.idx--
		return s.data[s.idx+1], true
	}
	return *new(T), false
}

func (s *Stack[T]) Size() int {
	fmt.Println(s.idx)
	if s.idx == -1 {
		return 0
	}
	return s.idx
}

func (s *Stack[T]) String() string {

	if s.idx < 1 {
		return ""
	}

	b := strings.Builder{}

	for i := 0; i <= s.idx; i++ {
		if i == 0 {
			b.WriteString(fmt.Sprintf("%v", s.data[i]))
		} else {
			b.WriteString(", " + fmt.Sprintf("%v", s.data[i]))
		}
	}

	return b.String()
}