基础知识:
1.比较两个数的大小min和maxc++ 使用 algorithm 提供的 max/min,返回最小最大的数。
#include 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#include #include //比大小头文件 class MinStack { std::stack x_stack; std::stack 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; }
运行结果



