括号匹配问题
栈的典型题目还是非常明显的,括号匹配、表达式计算等等几乎都少不了栈,本小节我们就看两个最经典的问题。
LeetCode20-有效的括号
给定一个只包括 '(',')','{','}','[',']' 的字符串 s ,判断字符串是否有效。
有效字符串需满足:
左括号必须用相同类型的右括号闭合。
左括号必须以正确的顺序闭合。
每个右括号都有一个对应的相同类型的左括号。
示例 1:
输入:s = "()"
输出:true
示例 2:
输入:s = "()[]{}"
输出:true
示例 3:
输入:s = "(]"
输出:false
本题还是比较简单的,其中比较麻烦的是如何判断两个符号是不是一组的,我们可以用哈希表将所有符号先存储,左半边做key,右半边做value。遍历字符串的时候,遇到左半边符号就入栈,遇到右半边符号就与栈顶的符号比较,不匹配就返回false
代码如下:
public boolean isValid(String s) {
Map<Character, Character> hashMap = new HashMap<>();
hashMap.put('(', ')');
hashMap.put('[', ']');
hashMap.put('{', '}');
if (s.length() <= 1) {
return false;
}
Stack<Character> stack = new Stack<>();
for (int i = 0; i < s.length(); i++) {
char item = s.charAt(i);
if (hashMap.containsKey(item)) {
stack.push(item);
} else {
if (!stack.isEmpty()) {
Character left = stack.pop();
Character right = hashMap.get(left);
if (right != item){
return false;
}
} else {
return false;
}
}
}
return stack.isEmpty();
}
最小栈
LeetCode 155,设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
实现 MinStack 类:
MinStack() 初始化堆栈对象。
void push(int val) 将元素val推入堆栈。
void pop() 删除堆栈顶部的元素。
int top() 获取堆栈顶部的元素。
int getMin() 获取堆栈中的最小元素。
示例 1:
输入:
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]
输出:
[null,null,null,null,-3,null,0,-2]
解释:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
本题的关键在于理解getMin()到底表示什么,可以看一个例子上面的示例画成示意图如下:
这里的关键是理解对应的Min栈内,中间元素为什么是-2,理解了本题就非常简单。
题目要求在常数时间内获得栈中的最小值,因此不能在 getMin() 的时候再去计算最小值,最好应该在 push 或者 pop 的时候就已经计算好了当前栈中的最小值。
对于栈来说,如果一个元素 a 在入栈时,栈里有其它的元素 b, c, d,那么无论这个栈在之后经历了什么操作,只要 a 在栈中,b, c, d 就一定在栈中,因为在 a 被弹出之前,b, c, d 不会被弹出。
因此,在操作过程中的任意一个时刻,只要栈顶的元素是 a,那么我们就可以确定栈里面现在的元素一定是 a, b, c, d。
那么,我们可以在每个元素 a 入栈时把当前栈的最小值 m 存储起来。在这之后无论何时,如果栈顶元素是 a,我们就可以直接返回存储的最小值 m。
按照上面的思路,我们只需要设计一个数据结构,使得每个元素 a 与其相应的最小值 m 时刻保持一一对应。因此我们可以使用一个辅助栈,与元素栈同步插入与删除,用于存储与每个元素对应的最小值。
- 当一个元素要入栈时,我们取当前辅助栈的栈顶存储的最小值,与当前元素比较得出最小值,将这个最小值插入辅助栈中;
- 当一个元素要出栈时,我们把辅助栈的栈顶元素也一并弹出;
在任意一个时刻,栈内元素的最小值就存储在辅助栈的栈顶元素中。
public class MinStack {
Deque<Integer> xStack;
Deque<Integer> minStack;
public MinStack() {
xStack = new LinkedList<>();
minStack = new LinkedList<>();
minStack.push(Integer.MAX_VALUE);
}
public void push(int val) {
xStack.push(val);
minStack.push(Math.min(minStack.peek(), val));
}
public void pop() {
xStack.pop();
minStack.pop();
}
public int top() {
return xStack.peek();
}
public int getMin() {
return minStack.peek();
}
}
最大栈
LeetCode 716.设计一个最大栈数据结构,既支持栈操作,又支持查找栈中最大元素。
实现 MaxStack 类:
MaxStack stack = new MaxStack();
stack.push(5);
stack.push(1);
stack.push(5);
stack.top(); ->5
stack.popMax(); ->5
stack.top(); ->1
stack.peekMax(); ->5
stack.pop(); ->1
stack.top(); ->5
本题与上一题的相反,但是处理方法是一致的。一个普通的栈可以支持前三种操作 push(x),pop() 和 top(),所以我们需要考虑的仅为后两种操作 peekMax() 和 popMax()。
对于 peekMax(),我们可以另一个栈来存储每个位置到栈底的所有元素的最大值。例如,如果当前第一个栈中的元素为 [2, 1, 5, 3, 9],那么第二个栈中的元素为 [2, 2, 5, 5, 9]。在 push(x) 操作时,只需要将第二个栈的栈顶和 xx 的最大值入栈,而在 pop() 操作时,只需要将第二个栈进行出栈。
对于 popMax(),由于我们知道当前栈中最大的元素值,因此可以直接将两个栈同时出栈,并存储第一个栈出栈的所有值。当某个时刻,第一个栈的出栈元素等于当前栈中最大的元素值时,就找到了最大的元素。此时我们将之前出第一个栈的所有元素重新入栈,并同步更新第二个栈,就完成了 popMax() 操作。
public class MaxStack {
Deque<Integer> xStack;
Deque<Integer> maxStack;
public MaxStack() {
xStack = new LinkedList<>();
maxStack = new LinkedList<>();
}
public void push(int val) {
xStack.push(val);
int max = maxStack.isEmpty() ? val : maxStack.peek();
maxStack.push(Math.max(max, val));
}
public int pop() {
maxStack.pop();
return xStack.pop();
}
public int top() {
return xStack.peek();
}
public int peekMax() {
return maxStack.peek();
}
public int popMax() {
int max = peekMax();
Stack<Integer> buffer = new Stack<>();
while (top() != max) {
buffer.push(pop());
}
pop();
while (!buffer.isEmpty()) {
push(buffer.pop());
}
return max;
}
}