基础知识:
1.比较两个数的大小min和max
c++ 使用 algorithm 提供的 max/min,返回最小最大的数。
#include<algorithm>
std::max(2, 3);
std::min(2, 3);
int temp = stack.top();
std::min(temp,2);
2.题目:包含min函数的栈
本题难点: 通过建立辅助栈实现;
数据栈 AA : 栈 AA 用于存储所有元素,保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常逻辑。
辅助栈 BB : 栈 BB 中存储栈 AA 中所有 非严格降序 的元素,则栈 AA 中的最小元素始终对应栈 BB 的栈顶元素,即 min() 函数只需返回栈 BB 的栈顶元素即可。
因此,只需设法维护好 栈 BB 的元素,使其保持非严格降序,即可实现 min() 函数的 O(1)O(1) 复杂度。
函数设计:
push(x) 函数: 重点为保持栈 BB 的元素是 非严格降序 的。
将 xx 压入栈 AA (即 A.add(x) );
若 ① 栈 BB 为空 或 ② xx 小于等于 栈 BB 的栈顶元素,则将 xx 压入栈 BB (即 B.add(x) )。
pop() 函数: 重点为保持栈 A, B 的 元素一致性 。
0与-2对比-2较小,所以再赋值一个-2给min_stack,当执行pop时,是将两个栈一起各弹出,这样的好处是再添加数的时候不用判断谁小就加,不小就不加,而是只要小就加,而且再pop时也不用判断只有当两个栈顶数相等才弹出栈B。
top() 函数: 直接返回栈 AA 的栈顶元素即可,即返回 A.peek() 。
min() 函数: 直接返回栈 BB 的栈顶元素即可,即返回 B.peek() 。
#include<iostream>
#include<stack>
#include<algorithm> //比大小头文件
class MinStack {
std::stack<int> x_stack;
std::stack<int> min_stack; //辅助栈专门用来存放最小的数
public:
MinStack() {
}
void push(int x) {
x_stack.push(x);
if (min_stack.empty()) //栈是空的就直接添加,不用对比
{
min_stack.push(x);
}
else
{
int temp = min_stack.top();
min_stack.push(std::min(temp, x));
}
}
void pop() {
x_stack.pop(); //两个一起弹出
min_stack.pop();
}
int top() {
return x_stack.top();
}
int min() {
return min_stack.top();
}
};
int main() {
MinStack a;
a.push(-2);
a.push(0);
a.push(-3);
a.push(1);
std::cout << a.min() << std::endl;
a.pop();
a.pop();
std::cout << a.min() << std::endl;
}
运行结果