盛最多水的容器

给你 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

Solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#include <iostream>
#include <vector>
#include <cmath>
using namespace std;

class Solution {
public:
    int maxArea(vector<int>& height) {
        int res = 0;
        int l = 0;
        int r = height.size() - 1;
        while (l < r) {
            res = max((r - l) * min(height[l], height[r]), res);
            if (height[l] < height[r]) {
                l++;
            } else {
                r--;
            }
        }
        return res;
    }
};

int main() {
    std::vector<int> a = {4,3,2,1,4};
    std::cout << Solution().maxArea(a) << std::endl; // 16
    a = {1, 1};
    std::cout << Solution().maxArea(a) << std::endl; // 1
    a = {1,2,1};
    std::cout << Solution().maxArea(a) << std::endl; // 2
}