剑指Offer第二十题：包含min函数的栈

题目描述

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

 

思路：唉，一开始写了个链表虽然能成功，但是还是太麻烦，这里应该直接用栈来做，STL是神器，直接利用两个stack，一个存数，一个存最小数。

存最小数的栈原理是，现存入一个数，第二数如果比top顶上的小，那么继续存入，存数字的栈弹出的时候，如果弹出的元素的值跟存最小数的top值相等，那么存最小数的栈也弹出。

代码如下

class Solution {
public:
    void push(int value) {
        va.push(value);
        if(myMin.empty())
            myMin.push(value);
        else if(value < myMin.top())
            myMin.push(value);
    }
    void pop() {
        if(!va.empty())
        {
            if(va.top() == myMin.top())
                myMin.pop();
            va.pop();  
        }               
    }
    int top() {
        return va.top();
    }
    int min() {
        return myMin.top();
    }
    
private:
    stack<int>va;
    stack<int>myMin;
};