这道题很简单,没什么好说的,但是我还是要为这道题单开个文章,因为这道题大部分速度很快的解法都是错的。
这是一个59ms的算法,看起来挺漂亮的对吧?
class MinStack {
int min = Integer.MAX_VALUE;
Stack<Integer> s = new Stack<>();
public void push(int x){
if(x<=min){
s.push(min);
min=x;
}
s.push(x);
}
public void pop(){
if(min==s.pop()) min=s.pop();
}
public int top(){
return s.peek();
}
public int getMin(){
return min;
}
}
这是一个56ms的算法,看起来好像也不错?
class MinStack {
int min;
Stack<Integer> stack;
/** initialize your data structure here. */
public MinStack() {
stack = new Stack<Integer>();
min = Integer.MAX_VALUE;
}
public void push(int x) {
if(x <= min){
stack.push(min);
min=x;
}
stack.push(x);
}
public void pop() {
if(stack.pop() == min) min=stack.pop();
}
public int top() {
return stack.peek();
}
public int getMin() {
return min;
}
}
大哥们,这种重复存储第二小元素的的算法都是错的啊。只不过leetcode的测试代码有些naive,没测试出来罢了。
为什么是错的呢?考虑这种情况:push(5) push(4) push(6) push(4)。 你可以在纸上画一下,发现问题了吗?
5只被push了一次,但是在栈中却存储了两次,这类算法只要多push几个最小值就是错的,我不知道为什么这么明显的错误都没测试出来。。。