题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
题解
题目抽象:要求实现一个O(1)时间复杂度的返回最小值的栈。正常情况下,栈的push,pop操作都为O(1),
但是返回最小值,需要遍历整个栈,时间复杂度为O(n),所以这里需要空间换时间的思想。
方法:使用辅助栈
首先需要一个正常栈normal,用于栈的正常操作,然后需要一个辅助栈minval,专门用于获取最小值,具体操作如下。
- push操作就如图片上操作。
- pop操作直接对应弹出就好了。
- top操作就返回normal的栈顶
- min操作就返回minval的栈顶
class Solution {
public:
stack<int> normal, minval;
void push(int value) {
normal.push(value);
if(minval.empty())
minval.push(value);
else{
if(value <= minval.top())
minval.push(value);
else
minval.push(minval.top());
}
}
void pop() {
normal.pop();
minval.pop();
}
int top() {
return normal.top();
}
int min() {
return minval.top();
}
};