刷题笔记(十七)——包含min函数的栈
一直在忙着复现论文,好久不刷题了,论文笔记也有几篇没更,近期补上~
题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
思路:
在栈中加入一个可以输出当前栈中最小的元素函数
可以借助另一栈来存放当前栈中最小的元素,栈1进行push操作时,若该元素小于等于栈2中栈顶元素,则栈2也执行push操作。栈1进行pop操作时,若栈1的栈顶元素即要pop的元素等于栈2 中的栈顶元素,则栈2也执行pop操作,否则,栈2不进行pop操作。
则栈1中的最小元素即为栈2的栈顶元素。
代码
class Solution {
public:
stack<int> s1;
stack<int> s2;
void push(int value) {
s1.push(value);
if(s2.empty())
s2.push(value);
else if(s2.top()>=value)//注意等号
s2.push(value);
}
void pop() {
if(s1.top()==s2.top())
{
s1.pop();
s2.pop();
}
else
s1.pop();
}
int top() {
return s1.top();
}
int min() {
return s2.top();
}
};