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

题目：包含min函数的栈

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

class Solution {
public:
    void push(int value) {
        
    }
    void pop() {
        
    }
    int top() {
        
    }
    int min() {
        
    }
};

 

解题代码：

class Solution {
public:
    void push(int value) {
        if(stack1.empty())
            stack2.push(value);
        else
            if(value < stack2.top())
                stack2.push(value);
            else
                stack2.push(stack2.top());

        stack1.push(value);
    }

    void pop() {
        stack1.pop();
        stack2.pop();
    }

    int top() {
        return stack1.top();
    }

    int min() {
        return stack2.top();
    }
private:
    stack<int> stack1;
    stack<int> stack2;
};