剑指 Offer 30. 包含min函数的栈(辅助栈)

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.

提示：

各函数的调用总次数不超过 20000 次

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof

解题思路：

考虑入栈，根据栈的特性，先入栈的元素沉到栈底，先进后出
考虑用一个辅助栈来保存栈中元素的最小值，原栈每入栈一个元素，就判断一下当前辅助栈的栈顶元素和入原栈元素的大小关系，如果要入栈的元素相等（重复元素入栈）或者更小，就同时把这个元素压入原栈和辅助栈
考虑出栈，如果出栈元素等于辅助栈栈顶（Integer引用类型，用equals判断），则辅助栈也要出栈

class MinStack {

    Stack<Integer>stack1=new Stack<>();
    Stack<Integer>stack2=new Stack<>();
    /** initialize your data structure here. */
    public MinStack() {     
    }
    
    public void push(int x) {
     stack1.push(x);
     if(stack2.empty())
     {
         stack2.push(x);
     }
     else
     {
         if(stack2.peek()>=x)
         {
             stack2.push(x);
         }
     }
    }
    
    public void pop() {
        if(!stack1.empty())  
        {
          if( stack1.pop().equals(stack2.peek()))
          {
              stack2.pop();
          }                  
           
        }
    }
    
    public int top() {
        return stack1.peek();
    }
    
    public int min() {
       return stack2.peek();
    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.min();
 */

解决
这样stack2是无法出栈的，靠，搞了好久。。。。

if( stack1.pop()==stack2.peek())
          {
              stack2.pop();
          }                

应该改成

if( stack1.pop().equals(stack2.peek()))
          {
              stack2.pop();
          }           

参考以及推荐

面试题30. 包含 min 函数的栈（辅助栈，清晰图解）
Java中Integer和int之间的转换
Java基础面试题3：==和equals的区别

总结

1：栈里面的元素类型是Integer我们比较大小的时候就要用equals来比较，因为对于引用类型==比较的是地址值，所以即使值相等，也会返回false导致stack2无法弹出栈顶元素

2：可以把Integer转成int来进行比较

 if( stack1.pop().intValue()==stack2.peek().intValue())
          {
              stack2.pop();
          }