(1) 每次弹出栈顶能保证min,因此压入栈的时候我们队栈里面的元素做排序操作,使得最小的元素处于栈顶的位置, 但是这样不能保证栈的先入后出的规则
(2) 对于时间复杂度有一定的要求,那么我们可以考虑以空间换时间的思想
#include<iostream>
#include<stack>
using namespace std;
//包含min函数的栈
class Solution {
public:
void push(int value) {
m_data.push(value);
if(m_temp.empty() || value < m_temp.top()){
m_temp.push(value);
}else{
m_temp.push(m_temp.top());
}
}
void pop() {
if(!m_data.empty() && !m_temp.empty()){
m_data.pop();
m_temp.pop();
}
}
int top() {
if(!m_data.empty()){
return m_data.top();
}
}
int min() {
if(!m_temp.empty()){
return m_temp.top();
}
}
public:
stack<int> m_data;//源数据栈
stack<int> m_temp;//辅助栈
};
//for test
int main(){
Solution s;
s.push(3);
s.push(4);
s.push(2);
s.push(1);
s.pop();
s.pop();
s.min();
int l = s.m_data.size();
for(int i =0;i<l;i++){
cout<<s.m_data.top();
s.m_data.pop();
}
for(int i =0;i<l;i++){
cout<<s.m_temp.top();
s.m_temp.pop();
}
return 0;
}