剑指offer20 包含min函数的栈

问题描述

描述
定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数，并且调用 min函数、push函数 及 pop函数 的时间复杂度都是 O(1)
push(value): 将value压入栈中
pop(): 弹出栈顶元素
top(): 获取栈顶元素
min(): 获取栈中最小元素

示例:
输入: [“PSH-1”,“PSH2”,“MIN”,“TOP”,“POP”,“PSH1”,“TOP”,“MIN”]
输出: -1,2,1,-1

解析:
"PSH-1"表示将-1压入栈中，栈中元素为-1
"PSH2"表示将2压入栈中，栈中元素为2，-1
“MIN”表示获取此时栈中最小元素==>返回-1
"TOP"表示获取栈顶元素==>返回2
"POP"表示弹出栈顶元素，弹出2，栈中元素为-1
"PSH-1"表示将1压入栈中，栈中元素为1，-1
"TOP"表示获取栈顶元素==>返回1
“MIN”表示获取此时栈中最小元素==>返回-1

测试用例：
输入： [“PSH-1”,“PSH2”,“MIN”,“TOP”,“POP”,“PSH1”,“TOP”,“MIN”]
返回值： -1,2,1,-1

解题思路

建立辅助栈，存放栈中最小值

第一次读题时下意识在min()中做大小排序，再次读题发现要求时间复杂度为O(1)，所以要在push中实现min，但是输入至栈中后顺序不变，只不过把栈中的最小值放在了栈顶位置，因此，就引进一个minstack栈，在每次push时，判断value与minstack栈顶的大小关系，始终把栈中的最小值放在minstack栈顶部分。

时间复杂度：O(1)
空间复杂度：O(n)

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

};