面试题21：包含min函数的栈

最新推荐文章于 2017-05-30 19:09:20 发布

原创最新推荐文章于 2017-05-30 19:09:20 发布 · 292 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

52 篇文章

订阅专栏

本文介绍了一种在栈数据结构中实现获取最小元素的方法。通过维护两个栈：一个用于存储数据元素，另一个用于跟踪最小值。这种方法确保了push、pop和min操作的时间复杂度均为O(1)。

题目：定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的min函数，调用min,push及pop的时间复杂度为O(1)。

这个题目push和pop都好做，但是min感觉不好做。思考可以用一个临时量记录min的值。但是没想到用一个数据结构去保存每次入栈、出栈的最小值。哎。。。

看了下原文，是用另外一个栈去维护最小值，好吧。。。代码如下：

class Solution {
public:
    void push(int value) {
        dataStack.push(value);
        if(minStack.size()==0||value<minStack.top())
        {
            minStack.push(value);
        }
        else
        {
            minStack.push(minStack.top());
        }
    }
    void pop() {
        if(!dataStack.empty())
        {
            dataStack.pop();
            minStack.pop();
        }
    }
    int top() {
        return dataStack.top();
    }
    int min() {
        return minStack.top();
    }
private:
    stack<int> dataStack;
    stack<int> minStack;
};