栈:golang实现最小栈逻辑

题目介绍：设计一个支持 push ，pop ，top 操作，并能在常数时间内检索到最小元素的栈。

实现 MinStack 类:

• MinStack() 初始化堆栈对象。
• void push(int val) 将元素val推入堆栈。
• void pop() 删除堆栈顶部的元素。
• int top() 获取堆栈顶部的元素。
• int getMin() 获取堆栈中的最小元素。

代码实现逻辑

package main

import (
    "errors"
    "fmt"
    "testing"
)

// MinStack 定义一个最小栈的结构体
type MinStack struct {
    stack    []int
    minStack []int
}

// NewMinStack 创建一个新的最小栈
func NewMinStack() *MinStack {
    return &MinStack{
       stack:    []int{},
       minStack: []int{},
    }
}

// Push 向栈中压入元素
func (s *MinStack) Push(x int) {
    s.stack = append(s.stack, x)
    // 如果 minStack 为空，或者 x 小于等于 minStack 栈顶元素，则压入 minStack
    if len(s.minStack) == 0 || x <= s.minStack[len(s.minStack)-1] {
       s.minStack = append(s.minStack, x)
    }
}

// Pop 从栈中弹出元素
func (s *MinStack) Pop() {
    if len(s.stack) == 0 {
       return
    }
    // 如果 stack 栈顶元素等于 minStack 栈顶元素，则同时弹出
    if s.stack[len(s.stack)-1] == s.minStack[len(s.minStack)-1] {
       s.minStack = s.minStack[:len(s.minStack)-1]
    }
    s.stack = s.stack[:len(s.stack)-1]
}

// Top 获取栈顶元素
func (s *MinStack) Top() (int, error) {
    if len(s.stack) == 0 {
       return 0, errors.New("stack is empty")
    }
    return s.stack[len(s.stack)-1], nil
}

// GetMin 获取栈中的最小值
func (s *MinStack) GetMin() (int, error) {
    if len(s.minStack) == 0 {
       return 0, errors.New("min stack is empty")
    }
    return s.minStack[len(s.minStack)-1], nil
}

func TestMinStack(t *testing.T) {
    minStack := NewMinStack()

    minStack.Push(3)
    minStack.Push(5)
    top, _ := minStack.Top()
    fmt.Printf("Top element: %d\n", top) // 输出 5

    min, _ := minStack.GetMin()
    fmt.Printf("Min element: %d\n", min) // 输出 3

    minStack.Push(2)
    minStack.Push(1)
    min, _ = minStack.GetMin()
    fmt.Printf("Min element after pushing 1: %d\n", min) // 输出 1

    minStack.Pop()
    min, _ = minStack.GetMin()
    fmt.Printf("Min element after popping: %d\n", min) // 输出 2

    minStack.Pop()
    top, _ = minStack.Top()
    fmt.Printf("Top element after popping twice: %d\n", top) // 输出 5

    min, _ = minStack.GetMin()
    fmt.Printf("Min element after popping twice: %d\n", min) // 输出 3
}