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

最新推荐文章于 2022-12-06 21:48:34 发布

原创最新推荐文章于 2022-12-06 21:48:34 发布 · 316 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer #c++

C++ 同时被 2 个专栏收录

81 篇文章

订阅专栏

剑指offer

25 篇文章

订阅专栏

本文介绍两种实现含min函数的栈的方法：一种使用两个栈分别存储数据和最小值；另一种利用vector实现相同功能。

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的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;
};