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

题目描述

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

时间限制：1秒 空间限制：32768K 热度指数：219343
本题知识点： 栈

思路

新建两个vector，一个当栈使用，一个存放对应栈位置的最小元素

class Solution {
public:
    void push(int value) {
        if(a.empty())
        {
            a.push_back(value);
            minc.push_back(value);
        }
        a.push_back(value);
        if(value<minc.back())
            minc.push_back(value);
        else minc.push_back(minc.back());
    }
    void pop() {
        int c=a.back();
        if(!a.empty())
        {
            a.pop_back();
            minc.pop_back();
        }
    }
    int top() {
        return a.back();
    }
    int min() {
        return minc.back();
    }
private:
    vector<int> a;
    vector<int> minc;
};

stack版本

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