剑指off day1
提示:这里可以添加系列文章的所有文章的目录,目录需要自己手动添加
例如:第一章 Python 机器学习入门之pandas的使用
提示:写完文章后,目录可以自动生成,如何生成可参考右边的帮助文档
前言
题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
提示:该题目是一个简单题,注意栈的初始化
示例:
> MinStack minStack = new MinStack();
> minStack.push(-2);
> minStack.push(0);
> minStack.push(-3);
> minStack.min(); --> 返回 -3.
> minStack.pop();
> minStack.top(); --> 返回 0.
> minStack.min(); -->返回 -2.
一、栈的初始化
class CQueue {
Deque<Integer> stack1;
Deque<Integer> stack2;
public CQueue() {
stack1 = new LinkedList<Integer>();
stack2 = new LinkedList<Integer>();
}
另一种初始化
Stack<Integer> A, B;
/** initialize your data structure here. */
public MinStack() {
A = new Stack<>();
B = new Stack<>();
二.算法解答
-
本题难点: 将 min() 函数复杂度降为 O(1)O(1) ,可通过建立辅助栈实现;
-
数据栈 AA : 栈 AA 用于存储所有元素,保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常逻辑。
-
辅助栈 BB : 栈 BB 中存储栈 AA 中所有 非严格降序 的元素,则栈 AA 中的最小元素始终对应栈 BB 的栈顶元素,即 min() 函数只需返回栈 BB 的栈顶元素即可。
-
因此,只需设法维护好 栈 BB 的元素,使其保持非严格降序,即可实现 min() 函数的 O(1)O(1) 复杂度。
函数设计:
-
push(x) 函数: 重点为保持栈 BB 的元素是 非严格降序 的。
- 将 xx 压入栈 A (即 A.add(x) );
- 若 ① 栈 B 为空 或 ② xx 小于等于 栈 B 的栈顶元素,则将 xx 压入栈 B (即 B.add(x) )。
-
pop() 函数: 重点为保持栈 A, B 的 元素一致性 。
1.执行栈 AA 出栈(即 A.pop() ),将出栈元素记为 yy ;
2.若 yy 等于栈 BB 的栈顶元素,则执行栈 B 出栈(即 B.pop() )。 -
top() 函数: 直接返回栈 AA 的栈顶元素即可,即返回 A.peek() 。
-
min() 函数: 直接返回栈 BB 的栈顶元素即可,即返回 B.peek() 。
代码示例:
class MinStack {
Stack<Integer> A, B;
public MinStack() {
A = new Stack<>();
B = new Stack<>();
}
public void push(int x) {
A.add(x);
if(B.empty() || B.peek() >= x)
B.add(x);
}
public void pop() {
if(A.pop().equals(B.peek()))
B.pop();
}
public int top() {
return A.peek();
}
public int min() {
return B.peek();
}
}
三、总结(踩过的坑)
提示:这里对文章进行总结:
踩了两个坑
坑一
public void pop() {
if(A.pop().equals(B.peek())){
B.pop();
}
A.pop();
}
其中A.pop().equals(B.peek())已经实现了A的出栈,相当于A出了两次栈,结果如下
坑二
这里的没写等号:B.peek() >= x 避免了重复最小值被弹出
public void push(int x) {
A.add(x);
if(B.empty() || B.peek() > x)
B.add(x);
}
坑三
此题如果用==将会无法通过 Integer的equals重写过,比较的是内部value的值, ==如果在[-128,127]会被cache缓存,超过这个范围则比较的是对象是否相同