特殊的数据结构 -- 单调栈总结

x33g5p2x  于2021-11-21 转载在 其他  
字(6.2k)|赞(0)|评价(0)|浏览(387)

这里标注一下,本文参考于 《labuladong的算法小抄》

1. 思想讲解

1.1 单调栈思想

单调栈顾名思义栈中的元素在某一方面呈单调趋势,栈中存的可以是相应的数组值,也可以是相应的数组值的下标。但是平时的题目没有相关的提醒,所以有时候很难看出是单调栈,所以我们刷题时遇到具有明显的单调性需要留一个心眼

1.2 那我们是怎样维护这个栈的呢?

模板解析:

核心思想很简单,其实就是怎样维护这个栈。我们对每一个数组元素和栈顶比较大小(如果维护的是单调递增栈,那么该数组元素大于栈顶元素就入栈,如果是该数组元素是小于等于栈顶元素,那么就对栈内元素进行出栈操作。维护的其他的类型栈操作类似)。

(1)、单调递减栈 : 求的是下一个更大的元素

  1. for(int i = num.length - 1; i >= 0; i --){
  2. while(!s.isEmpty() && num[i] >= s.peek()){
  3. s.pop();
  4. }
  5. s.push(num[i]);
  6. }

(2)、单调递增栈 : 求的是下一个更小的元素

  1. for(int i = num.length - 1; i >= 0; i --){
  2. while(!s.isEmpty() && num[i] <= s.peek()){
  3. s.pop();
  4. }
  5. s.push(num[i]);
  6. }

(3)、维护的是一个单调递减栈

  1. for(int i = 0; i < num.length; i ++){
  2. while(!s.isEmpty() && num[i] > s.peek()){
  3. s.pop();
  4. }
  5. s.push(num[i]);
  6. }

(4)、 维护的是一个单调递增栈

  1. for(int i = 0; i < num.length; i ++){
  2. while(!s.isEmpty() && num[i] > s.peek()){
  3. s.pop();
  4. }
  5. s.push(num[i]);
  6. }

2. 例题讲解

2.1 接雨水

AC代码:

  1. public int trap(int[] height) {
  2. int ans = 0;
  3. Stack<Integer> s = new Stack<>();
  4. for(int i = 0; i < height.length; i ++){
  5. while(!s.isEmpty() && height[s.peek()] < height[i]){
  6. int index = s.pop();
  7. if(s.isEmpty()) break;
  8. int left = s.peek();
  9. int min = Math.min(height[i],height[left]);
  10. ans += (min - height[index]) * (i - left - 1);
  11. }
  12. s.push(i);
  13. }
  14. return ans;
  15. }

题目解析:

为什么该题可以使用单调栈?
分析题意可知,只有当呈字状的才能接住雨水,所以在一系列的单调不增的数中,突然出现了一个数比前面的数大的话,我们就需要对前面的数进行结算。所以可以使用单调栈。

结算时的第一种情况

图中所标注的是下标,栈中存放的也是下标。且该种情况接不到雨水

为什么在内层循环中需要设置以下代码?

  1. if(s.isEmpty()) break;

当第一个元素的下标已经进栈,第二个的元素大于栈顶元素进入内层循环。栈对栈顶元素进行pop(),并且此时栈为空,我们需要对后面的代码进行终止。所以设置以上代码。

结算时的第二种情况

图中所标注的是下标,且该种情况下的所接雨水为4

当程序运行到下图所示:

准备进栈的元素为下标为4的元素,此时它的值比栈顶元素的大。

执行以下代码:

  1. int index = s.pop();
  2. if(s.isEmpty()) break;
  3. int left = s.peek();
  4. int min = Math.min(height[i],height[left]);
  5. ans += (min - height[index]) * (i - left - 1);

此时的index3left2,求出来的雨水量为1

此时如下图所示:

如上图所示的栈中元素还要经过一次的对4的pop,因为求出的雨水量为0,所以此处省略。那么就是如上图所示了。

执行以下代码:

  1. int index = s.pop();
  2. if(s.isEmpty()) break;
  3. int left = s.peek();
  4. int min = Math.min(height[i],height[left]);
  5. ans += (min - height[index]) * (i - left - 1);

此时的index2left1,求出来的雨水量为3

所以此时的总雨水量为4

2.2 柱状图中最大的矩形

题目解析
这是一道在竞赛中算简单的,但在笔试面试中属于难的题目。其实想明白了也就很简单了。我们对数组中的每一个元素进行处理出左边和右边的下一个更小的元素的下标,那么这道题也就解决了。

那么为什么如上所说就解决了?

因为假如我们情况如图所示:

左右两边的比它高的不会影响它的最大面积的扩展,只有当是高度小于它的时候才会影响

找到左边第一个比它小的数

  1. int[] l = new int[heights.length];
  2. Stack<Integer> s1 = new Stack<>();
  3. for(int i = 0; i < heights.length; i ++){
  4. while(!s1.isEmpty() && heights[i] <= heights[s1.peek()]){
  5. s1.pop();
  6. }
  7. l[i] = s1.isEmpty() == true ? -1 : s1.peek();
  8. s1.push(i);
  9. }

为什么栈空就设置为-1

  1. l[i] = s1.isEmpty() == true ? -1 : s1.peek();

因为此时左边没有元素比它小,所以设为最左边的 -1

找到右边第一个比它小的数

  1. int[] r = new int[heights.length];
  2. Stack<Integer> s2 = new Stack<>();
  3. for(int i = heights.length - 1; i >= 0; i --){
  4. while(!s2.isEmpty() && heights[i] <= heights[s2.peek()]){
  5. s2.pop();
  6. }
  7. r[i] = s2.isEmpty() == true ? heights.length : s2.peek();
  8. s2.push(i);
  9. }

为什么栈空设为heights.length?

因为右边没有比它更小的元素了,所以设置为heights.length

相应的面积计算

  1. res,heights[i] * (r[i] - l[i] - 1);

AC代码

  1. class Solution {
  2. public int largestRectangleArea(int[] heights) {
  3. int[] l = new int[heights.length];
  4. Stack<Integer> s1 = new Stack<>();
  5. int[] r = new int[heights.length];
  6. Stack<Integer> s2 = new Stack<>();
  7. // 找到左边第一个比它小的数,组成一个下标数组
  8. for(int i = 0; i < heights.length; i ++){
  9. while(!s1.isEmpty() && heights[i] <= heights[s1.peek()]){
  10. s1.pop();
  11. }
  12. l[i] = s1.isEmpty() == true ? -1 : s1.peek();
  13. s1.push(i);
  14. }
  15. // 找到右边第一个比它小的数组成一个下标数组。
  16. for(int i = heights.length - 1; i >= 0; i --){
  17. while(!s2.isEmpty() && heights[i] <= heights[s2.peek()]){
  18. s2.pop();
  19. }
  20. r[i] = s2.isEmpty() == true ? heights.length : s2.peek();
  21. s2.push(i);
  22. }
  23. // 以数组每个点作为高更新res
  24. int res = 0;
  25. for(int i = 0; i < heights.length; i ++){
  26. res = Math.max(res,heights[i] * (r[i] - l[i] - 1));
  27. }
  28. return res;
  29. }
  30. }

3. 相关 LeetCode 题目

3.1 移掉 K 位数字

题目分析:
很明显,需要返回的是一个数值最小的数字,维护的是一个单调递增栈,就是需要注意的是它这个k个元素必须去除和要防止前导零的出现

AC代码:

  1. class Solution {
  2. public String removeKdigits(String num, int k) {
  3. // 由于k 是必须要去除的 k个元素
  4. if(num.length() <= k) return "0";
  5. Deque<Character> queue = new LinkedList<>();
  6. for(int i = 0; i < num.length(); i ++){
  7. char c = num.charAt(i);
  8. while(!queue.isEmpty() && queue.peekLast() > c){
  9. // 为什么此处的 是 > ?
  10. // 这里需要注意的是 维护的是一个 单调不减栈,因为可能出现后面一个和前面的栈中的一个元素相等,但是其实前面的一个位于前面,后面返回值都不会影响,所以此处设置为维护的是一个单调不减栈。
  11. if(k <= 0){
  12. break;
  13. }
  14. // 此处的break 应该很好理解
  15. queue.pollLast();
  16. k --;
  17. }
  18. queue.offerLast(c);
  19. }
  20. // 为什么此处要设置这一个while循环呢?
  21. // 因为题目要求必须去除k个元素,但是我们知道满足前面要求的不一定都除去过了k个元素了,
  22. // 比如 num = "112", k = 1;
  23. // 运行到此处是 sb = "112",显而易见有必要设置这样一个循环。
  24. while(k > 0){
  25. queue.pollLast();
  26. k --;
  27. }
  28. StringBuffer sb = new StringBuffer();
  29. while(!queue.isEmpty()){
  30. sb.append(queue.pollFirst());
  31. }
  32. while(sb.length() > 1 && sb.charAt(0) == '0'){
  33. sb.deleteCharAt(0);
  34. }
  35. return new String(sb);
  36. }
  37. }

注意点:

1. 这个k个元素必须去除

  1. // 由于k 是必须要去除的 k个元素
  2. if(num.length() <= k) return "0";

2. 为什么此处要设置这一个while循环呢?

  1. while(k > 0){
  2. queue.pollLast();
  3. k --;
  4. }

因为题目要求必须去除k个元素,但是我们知道满足前面要求的不一定都除去过了k个元素了比如 num = “112”, k = 1;运行到此处是 sb = “112”, 显而易见有必要设置这样一个循环

3. 为什么需要在内层循环中设置break?

因为最多去除k个字符

3.2 去除重复字母

AC代码:

  1. class Solution {
  2. public String removeDuplicateLetters(String s) {
  3. // 1.其实这题是单调栈的思想,但是我们为什么使用的是双端队列的这种数据结构呢?
  4. // 因为最后结果返回的是一个字符串,而需要进行前端出,但在处理过程中有需要进行后端出元素的操作,所以使用双端队列
  5. // 2.但为什么我们不能事先处理字符串呢?
  6. // 因为需要进行去重,但是我们又不能事先处理好字符串。因为我们不知道事先对那个多余的字符进行删除,而题目又要保证字符的相对顺序,所以不能事先去重。所以我们设置一个去重数组 inQueue[],判断是否在队列中。和回溯时的去重相似
  7. // 3.那为什么又要设置 count[] 数组呢?
  8. // 根据单调栈的套路,我们每一次需要对栈中的元素进行出栈,但是这仅仅是因为前面的字符大于等于后面要入栈的字符。而因为我们是从前往后遍历的s字符串,而当这个字符在我们后面的字符子串不再出现,而又将要出栈,我们要对此行为进行阻止。所以设置count[] 数组计数
  9. // 特别注意点
  10. // (1)、在元素进队列的时候,我们需要对 count[] 和 inQueue[] 中的相应值进行修改
  11. // (2)、就算队列中已经存在相应的元素了,我们也需要对 count[]中的数值修改,前已经说明了 count[]的含义,这里应该很好明白。
  12. Deque<Character> queue = new LinkedList<>();
  13. int[] count = new int[26];
  14. boolean[] inQueue = new boolean[26];
  15. for(int i = 0; i < s.length(); i ++){
  16. char c = s.charAt(i);
  17. count[c - 'a'] ++;
  18. }
  19. for(int i = 0; i < s.length(); i ++){
  20. char a = s.charAt(i);
  21. if(inQueue[a - 'a'] == true){
  22. count[a - 'a'] --;
  23. continue;
  24. }
  25. while(!queue.isEmpty() && queue.peekLast() >= a){
  26. if(count[queue.peekLast() - 'a'] == 0) break;
  27. char b = queue.pollLast();
  28. inQueue[b - 'a'] = false;
  29. }
  30. queue.offerLast(a);
  31. inQueue[a - 'a'] = true;
  32. count[a - 'a'] --;
  33. }
  34. StringBuffer sb = new StringBuffer();
  35. while(!queue.isEmpty()){
  36. sb.append(queue.pollFirst());
  37. }
  38. return new String(sb);
  39. }
  40. }

题目代码解析:

1. 其实这题是单调栈的思想,但是我们为什么使用的是双端队列的这种数据结构呢?

因为最后结果返回的是一个字符串,而需要进行前端出,但在处理过程中有需要进行后端出元素的操作,所以使用双端队列

2.但为什么我们不能事先处理字符串呢?

因为需要进行去重,但是我们又不能事先处理好字符串因为我们不知道事先对那个多余的字符进行删除,而题目又要保证字符的相对顺序,所以不能事先去重。 所以我们设置一个去重数组 inQueue[],判断是否在队列中和回溯时的去重相似

3.那为什么又要设置 count[] 数组呢?

根据单调栈的套路,我们每一次需要对栈中的元素进行出栈,但是这仅仅是因为前面的字符大于等于后面要入栈的字符。而因为我们是从前往后遍历的s字符串,而当这个字符在我们后面的字符子串不再出现,而又将要出栈,我们要对此行为进行阻止。所以设置count[] 数组计数

4.特别注意点

(1)、在元素进队列的时候,我们需要对count[] 和 inQueue[] 中的相应值进行修改

(2)、就算队列中已经存在相应的元素了,我们也需要对 count[]中的数值修改,前已经说明了 count[]的含义,这里应该很好明白。

相关文章