一、题目

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。

点击查看原题
难度级别:困难

二、思路

1)单调递减栈

根据暴力法的思路:遍历每个元素,针对每个元素又遍历之前的元素,只需要计算到大于当前元素的值即可(或没有大于的)。
可以使用一个单调递减栈,将递减序列记录下来,(想一下,就算两个递减序列值中间间隔很远,中间很多水,但已经计算过了,不需要重复计算)。只需要计算当前元素的存在,多留下多少水即可。具体考虑如下:

第一步:当前元素>栈顶元素,则出栈,(i - ``stack``.``peek``() - ``1``)*(height[``stack``.``peek``()] - height[temp])可以计算多留下的水,其中i是当前元素位置,temp是出栈元素最大值 第一步执行结束后,将当前元素添加进入栈中,那么,和栈顶元素又多留下的水量(i - ``stack``.``peek``() - ``1``)*(height[i] - height[temp])

2)双指针

使用两个指针分别为leftright,从左向右遍历和从右向左遍历过的最大值分别为maxLmaxR,当leftright的值不同时,移动小的那个指针,因为大的那个指针是达到新高点,必然能够截留另一侧比它低的雨水

三、代码

1)单调递减栈

  1. class Solution {
  2. public int trap(int[] height) {
  3. int ans = 0;
  4. Stack<Integer> stack = new Stack();
  5. for (int i = 0; i < height.length; i++) {
  6. int temp = i; // 使用temp记录最近pop出来的值
  7. while (!stack.isEmpty() && height[stack.peek()] <= height[i]) {
  8. ans += (i - stack.peek() - 1)*(height[stack.peek()] - height[temp]);
  9. temp = stack.pop(); // 由于是递减队列,新pop出来的一定是更大值
  10. }
  11. if (!stack.isEmpty()) {
  12. ans += (i - stack.peek() - 1)*(height[i] - height[temp]);
  13. }
  14. stack.push(i);
  15. }
  16. return ans;
  17. }
  18. }

时间复杂度为O(n),空间复杂度为O(n)

2)双指针

  1. class Solution {
  2. public int trap(int[] height) {
  3. int ans = 0;
  4. int left = 0, right = height.length - 1;
  5. int maxL = 0, maxR = 0;
  6. while (left < right) {
  7. maxL = Math.max(height[left], maxL);
  8. maxR = Math.max(height[right], maxR);
  9. if (height[left] < height[right]) {
  10. ans += maxL - height[left];
  11. left++;
  12. } else {
  13. ans += maxR - height[right];
  14. right--;
  15. }
  16. }
  17. return ans;
  18. }
  19. }

时间复杂度为O(n),空间复杂度为O(1)