题意:设计一个支持push / pop / top 以及常量时间检索最小元素的栈
- push(x) -- 将 x 放入栈顶
- pop() -- 移除栈顶元素
- top() -- 取得栈顶元素
- getMin() -- 获取栈中的最小元素
分析:
题目要求以常量时间取得栈中最小元素,所以在任意时刻你只能以常量时间得知当前栈中最小元素的位置。为了达到这个目的,我用一个栈(min_value_indexes)来记录 values 中最小元素的位置(即栈顶元素)。
操作:
用deque来模拟栈。每次 push 操作之前,将要入栈的元素和当前栈中的最小元素进行比较,如果要入栈的元素更小,则 min_value_indexes 将最小元素的位置(即values.size())入栈。然后 values 将 x 入栈。(第一次执行push 操作时,values 和 min_value_indexes 都为空,此时分别将 x 和 0放入各自栈中即可) 每次 pop 时,检查要移除的元素是否是当前栈中的最小元素,如果是,则 min_value_indexes.pop() 。
代码:
class MinStack {
public:
void push(int x) {
if (min_value_indexes.empty() || x < getMin())
{
min_value_indexes.emplace(values.size());
}
values.emplace_back(x);
}
void pop() {
values.pop_back();
if (min_value_indexes.top() == values.size())
{
min_value_indexes.pop();
}
}
int top() {
return values.back();
}
int getMin() {
return values[min_value_indexes.top()];
}
private:
std::deque<int> values;
std::stack<size_t> min_value_indexes;
};