OneBug-带最小值操作的栈

题目

实现一个栈, 支持以下操作:
push(val) 将 val 压入栈
pop() 将栈顶元素弹出, 并返回这个弹出的元素
min() 返回栈中元素的最小值
要求 O(1) 开销.

保证栈中没有数字时不会调用 min()

样例
输入:
push(1)
min()
push(2)
min()
push(3)
min()
输出:
1
1
1

思路

一开始我想的太简单了，只考虑到push操作，我想着如果只有O（1）的空间开销的话，我只需要在每次push新数时，与当前的最小值Miin进行比较，若新数更小，Min = new number；后来真正码代码时才考虑到当做pop操作时若只是pop出了比Min更大的数还好，若恰好pop出了最小的Min之后，做了min（）操作后，我怎么知道新的最小的数？因此本题需要注意的一个点是，当做pop操作时，如果当前栈内最小的数恰好被弹出时，要能 正确的找到新的最小的数。这样就需要两个栈来实现：一个数正常栈stk，另一个是有关最小数的栈Minstk；

代码实现

class MinStack
{
  private:
    stack<int> stk, minStk;

  public:
    MinStack()
    {
        // do intialization if necessary
    }

    /*
     * @param number: An integer
     * @return: nothing
     */
    void push(int number)
    {
        stk.push(number);
        if (minStk.empty() || number <= minStk.top())	//若即将压栈的数比原来minstk栈顶的数还小，
        {												//则新数压入minstk栈顶，原来栈顶的数自动做第二小的数，
            minStk.push(number);					//这样可以保证minstk栈内的数从栈顶到栈底按升序。
        }											//因此当pop出最小的数后，只需同步弹出minstk栈顶的数，
    }												//就可以得到新的最小的数

    /*
     * @return: An integer
     */
    int pop()
    {
        if (stk.top() == minStk.top()) 			//若弹出的恰好是当前栈内最小的数
        {										//两个栈同时弹出即可
            minStk.pop();
        }
        int ret = stk.top();
        stk.pop();
        return ret;
    }

    /*
     * @return: An integer
     */
    int min()
    {
        return minStk.top();			//minstk的栈顶即为stk栈最小的数
    }
};