【小白刷题之路Day29】leetcode155. 最小栈（优化很关键）

leetcode155. 最小栈 设计一个支持 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. 来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/min-stack 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

第一次提交试用等容量的辅助栈来做，很容易想到：

class MinStack { private: stack<int> s; stack<int> s_min; public: /** initialize your data structure here. */ MinStack() { } void push(int x) { s.push(x); if (!s_min.empty()){ s_min.push(min(s_min.top(), x)); } else s_min.push(x); } void pop() { s.pop(); s_min.pop(); } int top() { return s.top(); } int getMin() { return s_min.top(); } };

 

接着查看别人的代码，发现可以优化辅助数组的容量，由于最小栈一些元素重复，当新元素大于最小栈栈顶元素时，新元素不入栈；pop时若弹出元素大于最小栈栈顶元素时，最小栈栈顶元素也不弹出，很好的优化思路。

添加两个约束条件即可实现，也不难。我尝试优化掉重复元素进入最小栈的问题，想了一下觉得可以预计的成本范围内无法实现。

class MinStack { private: stack<int> s; stack<int> s_min; public: /** initialize your data structure here. */ MinStack() { } void push(int x) { s.push(x); if (s_min.empty() || x<=s_min.top()) s_min.push(x); } void pop() { if (s.top() == s_min.top()) s_min.pop(); s.pop(); } int top() { return s.top(); } int getMin() { return s_min.top(); } };

 

总结本题属于栈的基本题本题的优化思路很棒，收获不小。

 

转载于:https://www.cnblogs.com/ACStrive/p/11603901.html