leetcode刷题笔记——接雨水

题目

接雨水leetcode题目链接

题解

class Solution {
public:
/**
 * 双指针
 * p1从数组第一个元素往后遍历，p2从数组最后一个元素往前遍历
 * 核心思想是计算每个元素上方可以接的雨水的数量
 * 例如，数组[4,2,0,3,2,5]，
 * 元素可接雨水由其左右的数组最大值决定
 * 元素4上可接雨水为0，
 * 元素2左边数组最大值为4，右边数组最大值为0，接不到雨水，即可接雨水数量为0
 * 元素0上左边数组最大值为4，右边数组最大值为5，可接雨水数量为min(4,5)-0；
 * ......
*/
    int trap(vector<int>& height) {
        int p1 = 0;
        int p2 = height.size() - 1;
        int leftmax = 0;
        int rightmax = 0;
        int res = 0;
        while(p1 <= p2)
        {
            if(leftmax > rightmax)
            {
                res += max(rightmax - height[p2], 0);
                rightmax = max(rightmax, height[p2]);
                p2--;
                
            }
            else
            {
                res += max(leftmax - height[p1], 0);
                leftmax = max(leftmax, height[p1]);
                p1++;
                
            }
        }
        return res;
    }
};