LeetCode155:Min Stack

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

  • push(x) – Push element x onto stack.
  • pop() – Removes the element on top of the stack.
  • top() – Get the top element.
  • getMin() – Retrieve the minimum element in the stack.

Hide Tags Stack Data Structure

这道题是要设计一个能够在常数时间内执行push,pop,top,获取栈中最小元素getMin函数的栈。

前面三个操作是一个栈固有的功能,但是在stl提供的栈中要获取栈中最小的元素明显不能在常数时间复杂度内完成,这就是问题的关键。

可以使用两个栈来实现在常数时间内获取最小的元素,第一个栈记录插入的数据,第二个栈记录插入当前数据时栈中的最小元素,之所以需要一个栈来保存最小元素而不是使用一个变量来保存最小元素的原因是当最小元素被弹出栈时我们不知道接下来的最小元素是多少了。

在Discuss中看到很多人在实现pop和top操作时对栈是否为空进行了判断,感觉这个判断是需要的,不然数组为空时还执行top操作明显是不合适的,但是stl中的pop和top好像没有做这个判断,这里最开始我也没考虑到,以后考虑问题要周全一些。

runtime:28ms

class MinStack {
public:
    void push(int x) {
        if(minData.empty())
            minData.push(x);
        else
        {
            if(x<minData.top())
                minData.push(x);
            else
                minData.push(minData.top());
        }
        data.push(x);
    }

    void pop() {
        data.pop();
        minData.pop();
    }

    int top() {
        return data.top();
    }

    int getMin() {
        return minData.top();
    }

    private:
    stack<int> data;
    stack<int> minData;
};

你可能感兴趣的:(LeetCode155:Min Stack)