设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。
LeetCode 第155题 https://leetcode-cn.com/problems/min-stack/
主要想法是维护一个辅助栈,里面存的是比较小的那些元素。
存入第一个元素两个栈都需要存。
之后再存就需要判断了。如果新元素小于辅助栈的头,则两个栈都存入。否则只存入主栈。
删除的时候,主栈的头必删,辅助栈看情况。如果两个栈的头相等,就同时删去,不等,只删主栈的头。
这样,最后的top是主栈的头,getmin是辅助栈的头。
这里的细节主要有
存入的数据可能会有重复。所以如果存入的数字和辅助栈里的头相等(即之前已经存入过这个元素并且这个元素最小),那还是需要让两个栈都存入这个元素。因为在这种情况下,不同时存入,pop()这个方法会出现不同步。
代码如下:
public class MinStack {
public int minlen;
public int length;
public int[] list;
public int[] min;
/** initialize your data structure here. */
public MinStack() {
minlen = 0;
length = 0;
list = new int[1439070];
min = new int[1439070];
}
public void Push(int x) {
list[length]=x;
length++;
if (minlen == 0)
{
min[0]=x;
minlen++;
}
else if (x <= min[minlen-1])
{
min[minlen]=x;
minlen++;
}
}
public void Pop() {
if (list[length-1] == min[minlen-1])
{
min[minlen-1]=0;
minlen--;
}
list[length-1]=0;
length--;
}
public int Top() {
return list[length-1];
}
public int GetMin() {
return min[minlen-1];
}
}
虽然想法是栈,但是我这里主要是用数组实现的,所以代码的有些地方看起来有些奇怪。
比如开了一大块存储空间,比如栈的头不是0而是length-1……
但是从某种程度上来说,这个minstack就是一个栈了,因为它确实实现了栈的功能。如果改成了正常的栈,也是这样的想法和做法。