Leetcode: Min Stack

卡尔曼和玻尔兹曼谁曼

发布于 2019-01-25 14:44:59

6350

发布于 2019-01-25 14:44:59

文章被收录于专栏：给永远比拿愉快

题目： 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.

思路分析：用两个stack来维护这个MinStack结构。1个stack用来正常进行stack的push pop等操作。另外1个stack用来维护min.每次对stack进行pop或者push时，也对min_stack进行相应操作。

C++代码示例：

#include <stack>

using namespace std;

class MinStack 
{
private:
    stack<int> stk;
    stack<int> min;
public:
    void push(int x)
    {
        stk.push(x);
        if (min.empty())
        {
            min.push(x);
        }
        else
        {
            //注意这里是>=，我第一次用>结果报错了
            if (min.top() >= x)
            {
                min.push(x);
            }
        }
    }

    void pop()
    {
        if (stk.top() == min.top())
        {
            min.pop();
        }
        stk.pop();
    }

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

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

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2015年03月07日，如有侵权请联系 cloudcommunity@tencent.com 删除

http