题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
思路:用一个栈data保存数据,用另外一个栈min保存依次入栈最小的数
no代表此次不如栈。
入栈:每次入栈的时候,如果入栈的元素比mindata中的栈顶元素小或等于则入栈,否则不如栈。
出栈:每次出栈的时候,如果data的栈顶元素和mindata栈顶元素不相同,说明data栈顶元素不是整个栈的最小元素,所以直接删除即可,如果两栈顶元素相同,则将data和mindata的栈顶一起出栈即可。
top:return data.top();
min:return mindata.top();
class Solution {
stack<int> data;
stack<int> mindata;
public:
void push(int value) {
data.push(value);
if (mindata.empty())
mindata.push(value);
if (mindata.top() > value)
mindata.push(value);
}
void pop() {
if(data.top() == mindata.top())
mindata.pop();
data.pop();
}
int top() {
return data.top();
}
int min() {
return mindata.top();
}
};