题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
应用一个辅助栈。压入:如果A栈的压入比B栈顶部元素大,B栈不压。小于等于:AB栈同时压入。出栈:如果AB栈顶元素不等,A出,B不出。
class Solution {
public:
stack<int> s1;
stack<int> s2;
void push(int value) {
s1.push(value);
if(s2.empty())
{
s2.push(value);
}
else
{
if(value <= s2.top())
s2.push(value);
}
}
void pop() {
if(s2.top() == s1.top())
s2.pop();
s1.pop();
}
int top() {
return s1.top();
}
int min() {
if(!s2.empty())
return s2.top();
return 0;
}
};
方法二:
道理差不多
存储当前元素下的最小值
class Solution {
public:
typedef pair<int, int> pii;
stack<pii> s;
void push(int value) {
if(s.empty())
{
s.push(pii(value, value));
}
else
{
if(value <= s.top().second)
s.push(pii(value, value));
else
s.push(pii(value, s.top().second));
}
}
void pop() {
s.pop();
}
int top() {
return s.top().first;
}
int min() {
return s.top().second;
}
};