11. 盛最多水的容器

题目

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

说明：你不能倾斜容器，且 n 的值至少为 2。

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

示例

输入：[1,8,6,2,5,4,8,3,7]
输出：49

思路

贪心算法，一个矩形的面积受最短边的影响，所以每次计算完当前矩形的面积后，选较短的边右移得到新边，然后继续计算。

代码

class Solution {
public:
    int maxArea(vector<int>& height) {
        int res = -1;
        int lo = 0, hi = height.size() - 1;

        while ( lo < hi ) {
            int h = height[lo] > height[hi] ? height[hi] : height[lo];
            int tempArea = h * ( hi - lo );

            res = max( res, tempArea );

            if ( height[lo] > height[hi] )
                --hi;
            else
                ++lo;
        }

        return res;
    }
};