题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
实现思想
利用两个栈来实现,栈1用来当普通栈使用,另一个辅助栈用来存储当前栈中的最小值(也就是每次压入value的时候,将value与辅助栈的栈顶比较,如果小于等于栈顶,不仅将数据压入栈1,而且将数据也压入辅助栈)
算法实现
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() {
//当stack1最上面的与stack2最上面的相等时就将stack2也抛出
if(stack1.top() == stack2.top()) stack2.pop();
//stakc1不管怎样都要弹出
stack1.pop();
}
int top() {
return stack1.top();
}
int min() {
return stack2.top();
}
private:
stack<int> stack1;
stack<int> stack2;
};
运行截图
总结
没什么好总结的,劳动节已经过去 好多天了,祝大家劳动节后快乐!
要不还是总结一点吧:当遇到处理最小值和最大值的问题都可以这样处理,比如用最小值举例,让算法返回当前数据中的最小值,都可用一个辅助数组或者列表或者栈来存储最小值,返回最小值就是返回数据中的最后一个元素即可。