1- Description
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
2-Solution
注意这里强调了得到最小元素要求时间复杂度是O(1),所以可以考虑将最小的元素存储起来备用,这里采取的方式是创建一个辅助栈进行最小元素的存储。
下面就以连续压入3、 4、 2、 1之后接连两次弹出栈顶元素,再压入0这一过程进行简单分析。
- 首先进行入栈操作,两个栈均添加同一元素,之后只有当前元素小于等于辅助栈顶元素辅助栈才有元素入栈(比如②就没有进行入栈操作)
- 随后的出栈操作,只要辅助栈的栈顶元素小于等于当前元素则二者均应该弹出,否则只有主栈弹出
根据上述算法步骤,可以完成如下解决代码:
class Solution {
stack<int> mdata;
stack<int> mmin;
public:
void push(int value) {
mdata.push(value);
if(mmin.size() == 0 || value <= mmin.top()) mmin.push(value);//一个存入最小值的栈
}
void pop() {
if(mdata.top() == mmin.top()) mmin.pop();//只有当主栈和辅助栈栈顶元素相等时才弹出
mdata.pop();
}
int top() {
return mdata.top();
}
int min() {
return mmin.top();
}
};