题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数
因为栈结构先进后出的,如果我们只使用一个变量业保存最小值,是不可行的。我们可以构造一个辅助函数来记录每个位置上对应的最小值。剑指offer里面的原题要求实现的每个函数操作的时间都是O(1),所以每次操作的时候就要对最小值辅助结构进行更新,这样当获得最小值的时候就可以是O(1)复杂度。
class Solution {
private:
std::stack<int> val;
std::stack<int> minVal;
public:
void push(int value) {
val.push(value);
if(minVal.empty() || (value < minVal.top())){
minVal.push(value);
}
else{
minVal.push(minVal.top());
}
}
void pop() {
val.pop();
minVal.pop();
}
int top() {
return val.top();
}
int min() {
return minVal.top();
}
};
但是这里面并没有做一个处理,也就是,当我们的栈为空没有元素时,对其进行弹出和返回栈顶元素是不行的,最好在里面再加上一句
assert(val.size()>0 && minVal.size() > 0);
牛客网的编译环境里面没有包含assert的头文件,而且不能够自己添加,所以就没有做错误处理了。
另外一种方法是使用pair来减少一个stack的使用,少了一个类型。
#define MIN(a,b) (a >b ?b:a)
class Solution {
stack<pair<int, int>> val;
public:
void push(int value) {
val.push(pair<int, int>(value, val.empty()?value:MIN(value, min())));
}
void pop() {
val.pop();
}
int top() {
return val.top().first;
}
int min() {
return val.top().second;
}
};