[每日一题]27:最小栈

题目描述

设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。

  • push(x) —— 将元素 x 推入栈中。
  • pop() —— 删除栈顶的元素。
  • top() —— 获取栈顶元素。
  • getMin() —— 检索栈中的最小元素。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

解答代码:

class MinStack {
    stack<int> m_data;
    stack<int> m_min;
public:
    /** initialize your data structure here. */
    MinStack() {

    }
    
    void push(int x) {
        m_data.push(x);

        if(m_min.empty() || m_min.top() >= x){
            m_min.push(x);
        }
    }
    
    void pop() {
        if(m_data.top() == m_min.top()){
            m_min.pop();
        }

        m_data.pop();
    }
    
    int top() {
        return m_data.top();
    }
    
    int getMin() {
        return m_min.top();
    }
};

如有不同见解,欢迎留言讨论!!!

发布了152 篇原创文章 · 获赞 45 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/AngelDg/article/details/105376010