hello-algo/ko/codes/go/chapter_stack_and_queue/array_stack.go

// File: array_stack.go
// Created Time: 2022-11-26
// Author: Reanon ([email protected])

package chapter_stack_and_queue

/* 배열로 구현한 스택 */
type arrayStack struct {
	data []int // 데이터
}

/* 스택 초기화 */
func newArrayStack() *arrayStack {
	return &arrayStack{
		// 스택의 길이를 0, 용량을 16으로 설정한다
		data: make([]int, 0, 16),
	}
}

/* 스택의 길이 */
func (s *arrayStack) size() int {
	return len(s.data)
}

/* 스택이 비어 있는지 판단 */
func (s *arrayStack) isEmpty() bool {
	return s.size() == 0
}

/* 스택에 삽입 */
func (s *arrayStack) push(v int) {
	// 슬라이스는 용량이 자동으로 확장된다
	s.data = append(s.data, v)
}

/* 스택에서 꺼내기 */
func (s *arrayStack) pop() any {
	val := s.peek()
	s.data = s.data[:len(s.data)-1]
	return val
}

/* 스택 맨 위 원소 반환 */
func (s *arrayStack) peek() any {
	if s.isEmpty() {
		return nil
	}
	val := s.data[len(s.data)-1]
	return val
}

/* 출력에 사용할 Slice 반환 */
func (s *arrayStack) toSlice() []int {
	return s.data
}