剑指offer题解C++【20】包含min的栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

解题思路1

定义两个栈，一个栈保存数据，另一个保存新数据入栈后的最小数
pop时，两个栈都要pop;
则minstack()的栈顶即为最小元素。

代码

class Solution {
public:
    void push(int value) {
        mystack.push(value);
        if (!minstack.empty()){
            minstack.push(std::min(minstack.top(), value));
        }
        else{
            minstack.push(value);
        }
    }
    void pop() {
        mystack.pop();
        minstack.pop();
    }
    int top() {
        return mystack.top();
    }
    int min() {
        return minstack.top();
    }

private:
    stack<int> mystack;
    stack<int> minstack;

};

解题思路2

题目并没有要求一定要用stack，用vector也能实现该功能

代码

class Solution {
public:
    void push(int value) {
        mystack.push_back(value);

        if(minstack.empty())
            minstack.push_back(value);
        else if(value<minstack.back())
            minstack.push_back(value);
        else 
            minstack.push_back(minstack.back());
    }
    void pop() {
        mystack.pop_back();
        minstack.pop_back();
    }
    int top() {
        return mystack.back();
    }
    int min() {
        return minstack.back();
    }
private:
    vector<int> mystack;
    vector<int> minstack;
};