题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
解题思路:
准备两个栈,一个存放数据,另一个存放最小值
#include<stack>
class Solution {
public:
void push(int value) {
stack1.push(value);
if(stack2.empty()){
stack2.push(value);
}else{
int minNum = stack2.top();
if(minNum > value){
stack2.push(value);
}else{
stack2.push(minNum);
}
}
}
void pop() {
stack1.pop();
stack2.pop();
}
int top() {
return stack1.top();
}
int min() {
return stack2.top();
}
private:
stack<int> stack1;//数据栈
stack<int> stack2;//最小数字栈
};