题目描述:
设计一个支持 push
,pop
,top
操作,并能在常数时间内检索到最小元素的栈。
-
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
输入: ["MinStack","push","push","push","getMin","pop","top","getMin"] [[],[-2],[0],[-3],[],[],[],[]] 输出: [null,null,null,null,-3,null,0,-2] 解释: MinStack minStack = new MinStack(); minStack.push(-2); minStack.push(0); minStack.push(-3); minStack.getMin(); --> 返回 -3. minStack.pop(); minStack.top(); --> 返回 0. minStack.getMin(); --> 返回 -2. 来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/min-stack 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
要做出这道题目,首先要理解「栈」结构先进后出的性质。
要求的 push(x) 、pop() 、top() 其实用正常的栈就可实现,
而要求 getMin() 在常数时间检索栈中的最小元素,则需要我们改造栈结构。
所以,我们可以使用一个「辅助栈」,与数据栈同步插入与删除,用于存储与每个元素对应的最小值。
-
当一个元素要入栈时,我们取当前辅助栈的栈顶存储的最小值,与当前元素比较得出最小值,将这个最小值插入辅助栈中;
-
当一个元素要出栈时,我们把辅助栈的栈顶元素也一并弹出;
-
在任意一个时刻,栈内元素的最小值就存储在辅助栈的栈顶元素中。
//go type MinStack struct { data []int //数据栈,存数据 help []int //辅助栈,存最小值 } /** initialize your data structure here. */ func Constructor() MinStack { return MinStack{ data: []int{}, help: []int{math.MaxInt64}, } } func (this *MinStack) Push(x int) { // 数据栈 this.data = append(this.data, x) // 辅助栈 this.help = append(this.help, min(x, this.help[len(this.help)-1])) } func (this *MinStack) Pop() { if len(this.data) != 0 { this.data = this.data[:len(this.data)-1] this.help = this.help[:len(this.help)-1] } } func (this *MinStack) Top() int { return this.data[len(this.data)-1] } func (this *MinStack) GetMin() int { return this.help[len(this.help)-1] } func min(x, y int) int { if x < y { return x } return y }
地址:https://mp.weixin.qq.com/s/r-Dv5hcR5VAhskupJP4pYA