【leetcode】42-接雨水【C++】

题目如下:

【leetcode】42-接雨水【C++】_第1张图片

解题思路:

首先要知道两个柱子间的水量是由短柱子决定的。采用 左右双指针 的方法,同时分别维护一个左右的当前最大柱子值:

1)、如果left_max < right_max,那么从左边开始遍历,并更新left_max以及结果大小;

2)、如果left_max >= right_max,那么从右边开始遍历,并更新right_max以及结果大小。

算法如下:

  • 初始化 left 指针为 0,并且right 指针为 size-1
  • While left < right, do:
    • If height[left] < height [left]
      • If height[left] > left_max:更新left_max;
      • Else:累加left_max - height[left];
      • left = left + 1;
    • Else
      • If height[right] > right_max:更新right_max;
      • Else:累加right_max - height[right];
      • right = right - 1;

代码如下:

class Solution {
public:
    int trap(vector& height) {
        int res = 0;
        int l_max = 0, r_max = 0; //维护左右最大值
        int l = 0, r = height.size() - 1; //左右“指针”
        while(l < r)
        {
            if(height[l] < height[r]){
                height[l] >= l_max ? l_max = height[l] : res += (l_max - height[l]);
                l++;
            }
            else{
                height[r] >= r_max ? r_max = height[r] : res += (r_max - height[r]);
                r--;
            }
        }
        return res;
    }
};

【leetcode】42-接雨水【C++】_第2张图片

你可能感兴趣的:(leetcode算法题库)