题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
自己的思路
由于考虑到时间复杂度要为O(1),所以就需要能直接取出最小值,试想,如果栈顶元素就是最小值时,就可以解决这个问题了,于是考虑使用两个栈,一个是最开始的入栈Stack1,不做处理,另一个栈Stack2每次通过比较,将第一个栈的最小值不断地压栈,最后将栈顶出栈就好了,但是可能是有条件考虑出错,导致不能a,先把错误代码贴出,后面再修改
class Solution {
public:
void push(int value) {
StackInt.push(value);
if(StackMin.empty())
StackMin.push(value);
else if(value<StackMin.top())
StackMin.push(value);
}
void pop() {
StackInt.pop();
}
int top() {
return StackInt.top();
}
int min() {
return StackMin.top();
}
private:
stack<int> StackInt;
stack<int> StackMin; //通过将StackMin的栈顶置为StackInt中的最小值,就可以是O(1)取出栈的最小值
};
大佬的思路
由于没有考虑出栈的情况,导致出错,所以还是要把出栈也考虑进去,不可能一直做压栈处理,那也太简单了,要保证每次出栈之后的栈顶元素始终是当前栈中的最小元素(两个栈的元素个数要相等)。(这也是解这道题的关键和细节所在)
通过一个策略,如果压栈的元素比StackMin的top元素大,则将StackMin的top元素再压栈一次,最终得到下面的两个栈,而每次出栈,只需要弹出栈顶元素就可以了,且StackMin的top永远是最小的,很妙。
class Solution {
public:
void push(int value) {
StackInt.push(value);
if(StackMin.empty())
StackMin.push(value);
else if(StackMin.top()<value)
StackMin.push(StackMin.top()); //为了保证两个栈的数据个数相等(这也是一个细节)
else
StackMin.push(value);
}
void pop() {
if(!StackInt.empty()){ //出栈的时候每次都将两个栈的栈顶出栈顶出栈就好了
StackInt.pop();
StackMin.pop();
}
}
int top() {
return StackInt.top();
}
int min() {
return StackMin.top();
}
private:
stack<int> StackInt;
stack<int> StackMin; //通过将StackMin的栈顶置为StackInt中的最小值,就可以是O(1)取出栈的最小值
};