包含min函数的栈

问题描述

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

思路

• 一个用来存所有的元素“total”,另一个用来存加入新的元素后当前total中对应的最小值。
• 两个栈中的元素数量始终保持一致，当新的元素小于“minValue”栈顶元素时，“minValue”向栈顶push新来的元素，否则，“minValue”向栈顶加入原栈顶元素。
• 执行“pop”方法时，两个栈同时弹出各自的栈顶元素。

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