首页 > 技术文章 > 包含min函数的栈

yl1995 2020-05-25 20:52 原文

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。

class Solution {
public:
void push(int value) {
  stack1.push(value);
  if(stack2.empty())
    stack2.push(value);
  if(value < stack2.top())
    stack2.push(value);
}
void pop() {
  if(stack1.top()==stack2.top())
    stack2.pop();
  stack1.pop();
}
int top() {
  return stack1.top();
}
int min() {
  return stack2.top();
}
private:
  stack<int> stack1;
  stack<int> stack2;
};

推荐阅读