接雨水问题

原题地址

接雨水

题目

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

示例 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 个单位的雨水（蓝色部分表示雨水）。 

解题思路

思路一

简单粗暴，计算每一点左右两边的最高峰，用两边中相对矮的一边减去当前坐标的高度，即可求出当前当前坐标的雨水量，每一个坐标点的雨水量相加就是所求。

int MaxhightL(int* hight,int i){
    int maxL=hight[i];
    for(int j=i-1;j>=0;j--){
        if(hight[j]>maxL){
            maxL=hight[j];
        }
    }
    return maxL;
}
int MaxhightR(int* hight,int i,int heightSize){
    int maxR=hight[i];
    for(int j=i+1;j<heightSize;j++){
        if(hight[j]>maxR){
            maxR=hight[j];
        }
    }
    return maxR;
}

int trap(int* hight, int heightSize){
    int maxL=0;
    int maxR=0;
    int sum=0;
    for(int i=1;i<heightSize;i++){
        maxL=MaxhightL(hight,i);
        maxR=MaxhightR(hight,i,heightSize);
        if(maxL>maxR){
            sum=sum+maxR-hight[i];
        }
        else{
            sum=sum+maxL-hight[i];
        }
    }
    return sum;
}

 思路二

先从左到右扫描一次，再从右向左扫描一次

将两次扫描所重叠的值就是所求的雨水值。

int trap(int* height, int heightSize) {
    int n = heightSize;
    if (n == 0) {
        return 0;
    }
    int leftMax[n];
    memset(leftMax, 0, sizeof(leftMax));
    leftMax[0] = height[0];
    for (int i = 1; i < n; ++i) {
        leftMax[i] = fmax(leftMax[i - 1], height[i]);
    }

    int rightMax[n];
    memset(rightMax, 0, sizeof(rightMax));
    rightMax[n - 1] = height[n - 1];
    for (int i = n - 2; i >= 0; --i) {
        rightMax[i] = fmax(rightMax[i + 1], height[i]);
    }

    int ans = 0;
    for (int i = 0; i < n; ++i) {
        ans += fmin(leftMax[i], rightMax[i]) - height[i];
    }
    return ans;
}