剑指offer——包含min函数的栈

题目描述：

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

本题知识点：

栈

解题思路：

应用一个辅助栈B，当压入栈A时，如果B为空则同时压入B中，如果B不为空，要压入的值如果小于B的top，则压入B中，否则不压入，从而保证B的top为A中当前压入所有值得最小值。当pop时，判断A所pop的值与B的top是否一致，如果一致则B也执行pop，否则不执行，从而保证B中的值一直为A中值得最小值，而不会因为A中当前的最小值pop后在B中仍然存在。

代码：

class Solution {
public:
    stack<int> s1, s2;
    void push(int value) {
        s1.push(value);
        if(s2.empty())
            s2.push(value);
        else if(value < s2.top())
            s2.push(value);
    }
    void pop() {
        if(s1.top() == s2.top())
            s2.pop();
        s1.pop();
    }
    int top() {
        return s1.top();
    }
    int min() {
        return s2.top();
    }
};