后面咱们学习了很多对于栈的常识,比方《动图演示:手撸堆栈的两种实现办法!》和《JDK 居然是这样实现栈的?》,那么接下来咱们再来刷一些对于栈的经典面试题以坚固学过的常识。
咱们明天的面试题是这样的 …
题目
定义栈的数据结构,请在该类型中实现一个可能失去栈的最小元素的 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.
LeetCode 地址:https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof/
思考
首先来说这道题目自身很好了解,它的实现难点在于以下两个方面:
- 当咱们进行 pop(移除栈顶元素)操作时如果删除的是以后最小值,那么咱们如何寻找下一个最小值?
- 要保障调用 min、push 及 pop 的工夫复杂度都是 O(1)。
也就是说,在咱们执行了 pop 时如果移除的栈中最小的值,那么如何寻找栈中的下一个最小元素?并且要保障操作的工夫复杂度为 O(1)。这个工夫复杂度制约了咱们在移除了最小值之后不能通过遍历查找下一个最小值,所以这就成为了这道题的难点。
比方当咱们移除以下栈顶元素值:
因为最小值就是 1,因而在移除之后最小值也被移除了,如下图所示:
那么接下来,让咱们一起思考 3 分钟,想一想应该如何解决这个问题~
解题思路
其实咱们能够 在每次入栈时,判断以后元素是否小于最小值,如果小于则将原最小值和最新的最小值相继入栈,这样在调用 pop 时即便移除的是最小值,咱们也能通过获取下一个元素失去一个新的最小值,执行流程如下所示。
操作步骤 1
入栈第一个元素,因为是第一个元素,因而最小值就是此元素的值。
操作步骤 2
入栈第二个元素,如下图所示:
因为入栈的元素 3 比 8 小,所以先将栈中的原最小值 8 存入栈中,再将 3 入栈。
操作步骤 3
入栈第三个元素,如下图所示:
因为入栈元素 5 大于 3,因而栈中的最小值不变,间接将元素 5 入栈。
操作步骤 4
持续入栈,如下图所示:
入栈元素 1 小于 3,因而先将原最小值 3 入栈,再将 1 入栈,栈中的最小值更改为 1。
操作步骤 5
执行出栈操作,如下图所示:
元素 1 出栈,判断以后元素就是栈的最小值,因而将栈顶元素 3 设置为最小值,并移除元素 3,如下图所示:
操作步骤 6
持续出栈,如下图所示:
因为元素 5 不是以后最小值,因而间接出栈。
操作步骤 7
持续出栈,如下图所示:
因为出栈元素 3 为最小值,因而持续将最小值设置为栈顶元素 8,并将栈顶元素出栈,如下图所示:
这样就剩下最初一个元素了,最初一个元素出栈之后就成空栈了,整个流程就执行完了。
实现代码 1
接下来咱们将下面的思路用代码实现一下,咱们用数组实现的栈来实现相干的性能,代码如下:
class MinStack {private int[] data; // 栈数据
private int maxSize; // 最大长度
private int top; // 栈顶指针(下标)private int min; // 最小值
// 构造函数
public MinStack() {
// 设置默认值
maxSize = 10000;
data = new int[maxSize];
top = -1;
min = Integer.MAX_VALUE;
}
// 入栈(增加元素)public void push(int x) {if (min >= x) {
// 遇到了更小的值,记录原最小值(入栈)data[++top] = min;
min = x;
}
// 以后值入栈
data[++top] = x;
}
// 出栈(移除栈顶元素)public void pop() {if (min == data[top]) {min = data[--top]; // 拿到原最小值,并(将原最小值)出栈
}
--top; // 出栈
}
// 查找栈顶元素
public int top() {return data[top];
}
// 查问最小值
public int min() {return min;}
}
上述代码在 LeetCode 的执行后果如下:
能够看出性能还是很高的,超过了 99.92% 的用户,内存耗费也不大。它的外围代码在 push
办法内,先将原最小值和最新最小值相继入栈,在 pop
出栈时判断出栈元素是否为最小值,如果是最小值则将以后最小值指向栈顶元素并将栈顶元素出栈,这样就失去了下一个新的最小值了。
实现代码 2
如果咱们不想应用数组的自定义栈来实现,还能够应用 Java 中自带的栈 Stack
来实现此性能,代码如下:
class MinStack {private Stack<Integer> stack = new Stack<>();
private int min = Integer.MAX_VALUE;
public MinStack() {}
// 入栈(增加元素)public void push(int x) {if (x <= min) {
// 遇到了更小的值,记录原最小值(入栈)stack.push(min);
min = x;
}
stack.push(x);
}
// 出栈(移除栈顶元素)public void pop() {if (stack.pop() == min) {min = stack.pop(); // 取出原最小值
}
}
// 查找栈顶元素
public int top() {return stack.peek();
}
// 查问最小值
public int min() {return min;}
}
上述代码在 LeetCode 的执行后果如下:
从后果能够看出,应用 Java 中自带的栈的性能不如自定义数组的栈,但代码还是通过了测试。这种实现形式的长处就是代码比较简单,能够利用了 Java 本身的 API 来实现了最小值的查找。
这种实现代码的形式(应用 Java API),在刷题或者理论面试中如果没有非凡阐明是能够间接用的。
总结
本文咱们通过两种形式:自定义数组栈和 Java API 中的 Stack
来实现了栈中最小值的性能,保障了在调用栈的 min、push 及 pop 办法时的工夫复杂度都是 O(1)。两种实现形式的代码尽管略不雷同,但实现思路都是一样的,都是 在元素入栈时判断以后元素是否小于最小元素,如果小于最小元素则先将原最小值入栈,再将以后最小元素入栈,这样当调用 pop 办法时,即便移除的是最小值,只须要将下一个元素取出即为新的最小值了,这样就能够实现调用 min、push 及 pop 办法时的工夫复杂度都是 O(1) 了。
最初
机智的你肯定还有其余的实现答案,评论区通知我吧~
原创不易,各位素质四连,谢啦。
文末福利:搜寻公众号「Java 中文社群」发送“面试”,支付最新的面试材料。