包含min函数的栈
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含的最小元素的min函数。(时间复杂度应为O(1))
注意:保证测试中不含当栈为空的时候对栈调用pop()或者top()或者min()函数。
题解:
对于此题,要保证时间复杂度为O(1)。即基本上对数据操作就是直接得到结果。
因此,采用额外的辅助栈来进行存储。
采用两个数组,第一个数组st1表示正常的数组,第二个数组st2则为辅助栈,用以保存每来一个数据或者输出数据时,st1中的最小值。
即当st1中只有一个数据时,即st1[0]不为空时,st2[0]存储的是st1[0];当st1中有两个数据时,st2[1]中存储的即为st1中两个数的最小值…随着数据增加,st2中的第i-1个位置,即为st1中i个数据中的最小元素。
这样,对st1、st2的操作完全相同,同时入栈一个元素,同时出栈一个元素。
代码如下:
class Solution {
private:
vector<int> st1;//声明两个数组
vector<int> st2;
public:
void push(int value) {
if(st2.empty())//当数组为空时,最小值即为刚刚入栈的元素
st2.push_back(value);
else{
if(value<st2.back())//当数组不为空时,最小值为st1中的最小值
st2.push_back(value);
else
st2.push_back(st2.back());//如果压入st1的数据大于之前的最小元素,则继续压入原来的最小元素
}
st1.push_back(value);
}
void pop() {
st1.pop_back();//同时出栈
st2.pop_back();
}
int top() {
return st1.back();
}
int min() {
return st2.back();
}
};