牛客网 | 高频面试题 | 容器盛水问题

题目

题目描述
给定一个整形数组arr，已知其中所有的值都是非负的，将这个数组看作一个容器，请返回容器能装多少水。
具体请参考样例解释
示例1
输入
复制
[3,1,2,5,2,4]
返回值
复制
5
说明

示例2
输入
[4,5,1,3,2]
返回值
2

解析

通过遍历求出arr[i]，左端和右端最大值；最后计算总容水量

class Solution {
public:
    /**
     * max water
     * @param arr int整型vector the array
     * @return long长整型
     */
    long long maxWater(vector<int>& arr) {
        // write code here
        int len = arr.size();
        long long res=0;
        vector<int> right(len),left(len);
        left[0]=arr[0],right[len-1]=arr[len-1];
        for(int i=1;i<len;++i){
            left[i]=max(left[i-1],arr[i]);
            right[len-i-1]=max(right[len-i],arr[len-i]);
        }
        for(int i=0;i<len;++i){
            res+=max(min(left[i],right[i])-arr[i],0);
        }
        return res;
    }
};