力扣热题100-双指针-字节跳动常见面试题：接雨水

题目描述：

题号：42

题目描述：

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

示例：

解题思路：

思路一：双指针

面对这道字节跳动面试常见的经典面试题，我们可以利用双指针的方法解决它。

算法思路：

1、两个指针 left 和 right 分别作为左右指针，两个变量 leftMax 和 rightMax记录左右指针遍历到的数字当中的最大中。初始值 left = 0, right = n − 1, leftMax = 0, rightMax = 0。左指针只会向右遍历，右指针只会向左遍历。

2、当 left < right 时，有两种情况（关键点:接水量由较矮的一边决定）：

情况1：

height[left] >= height[right]，则 leftMax > rightMax，则下标 left 处能接的雨水量等于 leftMax − height[left]，将下标 left 处能接的雨水量加到能接的雨水总量 answer 中，然后将 left 向右移动一位。

情况2:

height[left] < height[right]，则 leftMax < rightMax，则下标 right 处能接的雨水量等于 rightMax − height[right]，将下标 right 处能接的雨水量加到能接的雨水总量 answer 中，然后将 right 向左移动一位。

时间复杂度：O(N)

空间复杂度：O(1)

C++

// C++
class Solution {
public:
    int trap(vector<int>& height) {
        if(height.size() == 0) {
            return 0;
        }
        int answer = 0;
        int left = 0, right = height.size() - 1;
        int leftMax = 0, rightMax = 0;
        while(left <= right) {
            leftMax = max(leftMax, height[left]);
            rightMax = max(rightMax, height[right]);
            // 较矮的一边决定了能接多少水
            if(height[left] <= height[right]) {
                answer += leftMax - height[left];
                left++;
            } else {
                answer += rightMax - height[right];
                right--;
            }
        }
        return answer;
    }
};

go

// go
func trap(height []int) int {
    if len(height) == 0 {
        return 0
    }
    left := 0
    right := len(height) - 1
    leftMax, rightMax := 0, 0
    answer := 0
    for left < right {
        leftMax = max(leftMax, height[left])
        rightMax = max(rightMax, height[right])
        // 较矮的一边决定了能接多少水
        if height[left] <= height[right] {
            answer += leftMax - height[left]
            left++
        } else {
            answer += rightMax - height[right]
            right--
        }
    }
    return answer
}

func max(a, b int) int {
    if a > b {
        return a
    } else {
        return b
    }