题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
解题思路:
设一个辅助栈,用来存放最小值
入栈:
如果栈1的压入的值大于等于栈2,栈2不压;
如果栈1的压入的值小于栈2,栈1、栈2栈同时压入;
出栈:
如果栈1和栈2栈顶元素不相等,栈1出栈,栈2不出栈;
代码实现:
#include<stack>
class Solution {
/*
设一个辅助栈,用来存放最小值
入栈:
如果栈1的压入的值大于等于栈2,栈2不压;
如果栈1的压入的值小于栈2,栈1、栈2栈同时压入;
出栈:
如果栈1和栈2栈顶元素不相等,栈1出栈,栈2不出栈;
*/
stack<int> s1,s2;
public:
void push(int value) {
s1.push(value);
if(s2.empty()){
s2.push(value);
}else if(value < s2.top()){
s2.push(value);
}
}
void pop() {
if(s1.top() == s2.top()){
s2.pop();
}
s1.pop();
}
int top() {
return s1.top();
}
int min() {
return s2.top();
}
};