牛客題霸 [容器盛水問題] C++題解/答案

牛客題霸 [容器盛水問題] C++題解/答案

題目描述

給定一個整形數組arr，已知其中所有的值都是非負的，將這個數組看作一個容器，請返回容器能裝多少水。
具體請參考樣例解釋

題解：

我們找出容器的左右邊界，選擇邊界更低的，可以采用雙指針，分別從兩端向中間掃描，如果里面的高度高於邊界就不能盛水，更新邊界為當前高度，如果小於邊界就可以盛水，記錄答案

代碼：

class Solution {
public:
    /** * max water * @param arr int整型vector the array * @return long長整型 */
    long long maxWater(vector<int>& arr) {
        // write code here
        if(arr.size()<=2)return 0;
        int l=0,r=arr.size()-1;
        int L=arr[l];
        int R=arr[r];
        long long maxx=0;
        while(l<r)
        {
            if(L<R)
            {
                l++;
                if(arr[l]>=L)
                {
                    L=arr[l];
                    continue;
                }
                maxx+=(L-arr[l]);
            }
            else 
            {
                r--;
                if(arr[r]>R)
                {
                    R=arr[r];
                    continue;
                }
                maxx+=(R-arr[r]);
            }
        }
        return maxx;
    }
};