剑指offer-20包含min函数的栈

DF_0110

于 2018-08-15 11:28:39 发布

阅读量196

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/m0_37704296/article/details/81701223

本文介绍了一种使用两个栈实现能够在O(1)时间内获取栈内最小元素的方法。通过辅助栈记录每次压入主栈时的当前最小值，确保了在任何时候都能快速返回最小值。

题目描述
定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。
Note：寻找stack1中所含最小元素时，用另一个栈stack2存放读入的数据

class Solution {
public:

    void push(int value) {
        stack1.push(value);
    }
    void pop() {
        stack1.pop();
    }
    int top() {
        return stack1.top();
    }
    int min() {
        int minNum=0;
        if(!stack1.empty()){
            minNum=stack1.top();
            stack2.push(stack1.top());
            stack1.pop();
        }
        while(!stack1.empty()){
            if(minNum>stack1.top()){
                minNum=stack1.top();
                stack2.push(stack1.top());
                stack1.pop();
            }
            else{
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        while(!stack2.empty()){
            stack1.push(stack2.top());
            stack2.pop();
        }
        return minNum;
    }
private:
    stack<int> stack1;
    stack<int> stack2;
};