题目
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
实现 MinStack 类:
- MinStack() 初始化堆栈对象。
- void push(int val) 将元素val推入堆栈。
- void pop() 删除堆栈顶部的元素。
- int top() 获取堆栈顶部的元素。
- int getMin() 获取堆栈中的最小元素。
解法1:
使用两个栈来实现,一个存入数据dataStack,另一个来计算最小值的栈minStack,该方法可以避免再遍历一次堆栈,降低了时间复杂度,在很多堆栈问题中都会采用辅助栈的方式进行。
定义dataStack和minStack栈
type MinStack struct {
dataStack []int
minStack []int
}
push()
当时用push方法的时候,将会同时对dataStack和minStack进行入栈,dataStack直接放入,而对于minStack如果长度为0则直接放入,如果长度不为0则将栈顶部数据与存入数据进行比较,如果栈顶元素大于存入数据,则将存入数据压入堆栈,否则将保持不变。
func (this *MinStack) Push(val int) {
this.dataStack = append(this.dataStack, val)
n := len(this.minStack)
if n == 0 || this.minStack[n-1] > val {
this.minStack = append(this.minStack, val)
} else {
this.minStack = append(this.minStack, this.minStack[n-1])
}
}
pop()
使用pop()方法出栈就比较简单,对于dataStack和minStack都是将栈顶元素压出即可
func (this *MinStack) Pop() {
this.dataStack = this.dataStack[:len(this.dataStack)-1]
this.minStack = this.minStack[:len(this.minStack)-1]
}
top()
使用top()方法获取栈顶元素,只需要对dataStack操作即将dataStack的栈顶元素压出
func (this *MinStack) Top() int {
return this.dataStack[len(this.dataStack)-1]
Getmin()
使用Getmin()方法获取最小值元素,只需要对minStack操作即将minStack的栈顶元素压出
func (this *MinStack) GetMin() int {
return this.minStack[len(this.minStack)-1]
}