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

思路：

枚举所有高度（输入高度去重即可得到），按照这个高度找到最大的矩形，即需要找最左边大于或等于这个高度的一条边，再找到最右边大于或等于这个高度的一条边，这个高度和找到的两条边就形成了当前面积最大的矩形。

然后所有高度枚举出来的面积再求最大值即可得到答案。

算法复杂度是 n*log(n)

空间复杂度是O(n)

python AC代码：

class SegmentTree():
    def __init__(self, arr: List[int]):
        self.arr = arr
        self.n = len(arr)
        self.tree = [0 for i in range(self.n * 4)]
        self.build(1, self.n, 1)

    def build(self, l, r, rt):
        if l == r:
            self.tree[rt] = self.arr[l - 1]
            return

        m = (l + r) // 2
        self.build(l, m, rt * 2)
        self.build(m + 1, r, rt * 2 + 1)
        self.update_from_bottom(rt)

    def update_from_bottom(self, rt):
        self.tree[rt] = max(self.tree[rt << 1], self.tree[rt << 1 | 1])

    def query_left_greater_than_limit(self, l, r, rt, limit):
        if l == r:
            return l
        m = (l + r) >> 1
        if self.tree[rt << 1] >= limit:
            return self.query_left_greater_than_limit(l, m, rt << 1, limit)
        else:
            return self.query_left_greater_than_limit(m + 1, r, rt << 1 | 1, limit)

    def query_right_greater_than_limit(self, l, r, rt, limit):
        if l == r:
            return l
        m = (l + r) >> 1
        if self.tree[rt << 1 | 1] >= limit:
            return self.query_right_greater_than_limit(m + 1, r, rt << 1 | 1, limit)
        else:
            return self.query_right_greater_than_limit(l, m, rt << 1, limit)


class Solution:
    def maxArea(self, height: List[int]) -> int:
        tree = SegmentTree(height)
        uniq_height_set = set(height)
        ans = 0
        for h in uniq_height_set:
            if tree.tree[1] < h:
                continue
            l_index = tree.query_left_greater_than_limit(1, tree.n, 1, h)
            r_index = tree.query_right_greater_than_limit(1, tree.n, 1, h)
            ans = max(ans, (r_index - l_index) * h)
        return ans