【LeetCode热题100】【双指针】接雨水

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

示例 1:

【LeetCode热题100】【双指针】接雨水_第1张图片

输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出:6
解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 

示例 2:

输入:height = [4,2,0,3,2,5]
输出:9

提示:

  • n == height.length
  • 1 <= n <= 2 * 104
  • 0 <= height[i] <= 105

题解

这道题是双指针里面困难级别的题

我一开始的想法是用两个指针分别从左右两边出发,两边都是判断当前木板的高度是否低于先前碰到的最高的木板,如果是,那么累加二者的高度差,这样的思路基于一个前提:前面存在更高木板可以把水给罩住

但是存在一种情况,那就是一开始碰到的木板就是最高的,所以这种思路不行

官方给的思路是左右两边都计算一次,然后取二者间最小的

【LeetCode热题100】【双指针】接雨水_第2张图片

我在实现官方的思路的时候,想到了一种新的方法,一开始就去找到最高的那个木板所在的地方,仍然从左右两边出发去计算,但是碰到最高的地方我就停下来不算了

完美解决

【LeetCode热题100】【双指针】接雨水_第3张图片

class Solution {
public:
    int trap(vector &height) {
        int highest = 0;
        for (int i = 0; i < height.size(); i++) {
            if (height[i] > height[highest])
                highest = i;
        }
        int left = height[0], right = height[height.size() - 1], drop = 0, i = 1, j = height.size() - 2;
        while (i < highest) {
            if (height[i] < left) {
                drop += left - height[i];
            } else {
                left = height[i];
            }
            i++;
        }
        while (j>highest) {
            if (height[j] < right) {
                drop += right - height[j];
            } else {
                right = height[j];
            }
            j--;
        }
        return drop;
    }
};

你可能感兴趣的:(LeetCode热题,100,leetcode,算法,职场和发展)