LeetCode 11. 盛最多水的容器【二分】

给你 n 个非负整数 a1，a2，…，an，每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线，垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0) 。找出其中的两条线，使得它们与 x 轴共同构成的容器可以容纳最多的水。

说明：你不能倾斜容器。

示例 1：

输入：[1,8,6,2,5,4,8,3,7]
输出：49
解释：图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下，容器能够容纳水（表示为蓝色部分）的最大值为 49。

示例 2：

输入：height = [1,1]
输出：1

示例 3：

输入：height = [4,3,2,1,4]
输出：16

示例 4：

输入：height = [1,2,1]
输出：2

提示：

n = height.length
2 <= n <= 3 * 104
0 <= height[i] <= 3 * 104

枚举每一个柱子，找到其左边第一个大于它的柱子和右边第一个大于它的柱子。

class Solution {
    static const int maxn = 1e5 + 10;
    int lef[maxn];//存放左边第一个大于i的柱子
    int rig[maxn];//存放右边第一个大于i的柱子
    int lt = 0;
    int rt = 0;
public:
    int maxArea(vector<int>& height) {
        memset(lef, 0, sizeof(lef));
        memset(rig, 0, sizeof(rig));
        
        int len = height.size();
        for(int i = 0; i < len; i++) {
            if(lt == 0 || height[i] > height[lef[lt - 1]])
                lef[lt++] = i;//单调递增
        }
        for(int i = 0; i < len; i++) {
            if(rt == 0 || height[i] > height[rig[rt - 1]])
                rig[rt++] = i;
        }
        int ans = 0;
        for(int i = 0; i < len; i++) {
            int tem, l = 0, r = lt - 1;
            while(l <= r) {//二分查找第一个大于i的柱子
                int mid = l + r >> 1;
                if(height[lef[mid]] >= height[i])
                    tem = mid, r = mid - 1;
                else
                    l = mid + 1;
            }
            ans = max(ans, height[i] * (i - tem));//不断更新答案
        }
        for(int i = 0; i < len; i++) {
            int tem, l = 0, r = rt - 1;
            while(l <= r) {
                int mid = l + r >> 1;
                if(height[rig[mid]] >= height[i])
                    tem = mid, r = mid - 1;
                else
                    l = mid + 1;
            }
            ans = max(ans, height[i] * (tem - i));
        }
        return ans;
    }
};