当前位置: 代码迷 >> 综合 >> Leetcode-42.-Trapping-Rain-Water
  详细解决方案

Leetcode-42.-Trapping-Rain-Water

热度:96   发布时间:2023-12-16 19:55:40.0

title:Leetcode 42. Trapping Rain Water

1. 问题描述

Given n non-negative integers representing an elevation map where the
width of each bar is 1, compute how much water it is able to trap after raining.
Example:
Input: [0,1,0,2,1,0,1,3,2,1,2,1]
Output: 6

img

The above elevation map is represented by array [0,1,0,2,1,0,1,3,2,1,2,1]
In this case, 6 units of rain water (blue section) are being trapped. 

原题链接

2. 题解代码

class Solution {
public:int trap(vector<int>& A) {int left = 0; int right = A.size() - 1;int res = 0;int maxleft = 0, maxright = 0;while(left <= right){if(A[left] <= A[right]){if(A[left] >= maxleft) maxleft = A[left];else res += maxleft-A[left];left++;}else{if(A[right] >= maxright) maxright = A[right];else res += maxright-A[right];right--;}}return res;}
};

3. 代码思路分析

  • 首先这个图对分析问题很友好。可以看出,处于极大位置的height对于计算result起到关键作用。最终的result由所有相邻的两个极大之间能放的area之和构成
  • 注意到the width of each bar is 1,两极大之间各个barresult的贡献为min(左极大, 右极大) - 该bar的height
    is 1,两极大之间各个barresult的贡献为min(左极大, 右极大) - 该bar的height`
  • 一个简单的想法是,从左到右一路读过去标识出极大值,计算出答案。这里选择从左,右同时向中间读取,代码会更简洁(不过效率似乎差不多)