包含min函数的栈

题目描述

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

 

解题思路：

思路一：用一个vector来模拟栈，同时使用一个vmin来记录当前的最小值，不过当pop操作的时候，就需要重新遍历一遍vector中剩余的元素了，比较浪费时间。

思路二：使用两个stack、stack2来分别存放真实的栈数据和最小值。当入栈时，stack正常入栈，stack2判断当前入栈元素是否小于等于stack2栈顶元素，如果是就入栈stack2，否则不入stack2；在出栈时，如果stack栈顶元素等于stack2栈顶元素，则两个都出栈，否则只有stack出栈。（优秀的思路）

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