剑指offer 包含min函数的栈（C++）

最新推荐文章于 2020-03-24 12:09:47 发布

原创最新推荐文章于 2020-03-24 12:09:47 发布 · 199 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构

剑指offer 专栏收录该内容

68 篇文章

订阅专栏

本文介绍了一种在栈数据结构中实现O(1)时间复杂度获取最小元素的方法，通过维护辅助栈sMin来跟踪当前栈内的最小值，确保在不增加额外查询成本的情况下，快速获取栈内最小元素。

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为 $O (1)$ ）。
注意：保证测试中不会当栈为空的时候，对栈调用pop()或者min()或者top()方法。

解题思路

由于系统在栈空时不会调用类中的某些函数，所以无需对下列函数进行判空操作。

代码实现

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