問題の説明#
長さ n の整数配列 height が与えられます。n 本の垂直線があり、i 番目の線の 2 つの端点は (i, 0) と (i, height [i]) です。
その中から 2 本の線を見つけて、それらが 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
ヒント:
n == height.length
2 <= n <= 10^5
0 <= height[i] <= 10^4
考え方#
- 配列の両端から中央に向かって探索し、容器が保持できる水の上限は低い方の側に依存し、2 つのポインタの距離にこの値を掛けたものが水量になります。
- ポインタが移動するたびに、現在の水量と最大水量(初期値は 0)を比較し、最大値を取ります。
- 毎回片方のポインタだけを移動させ、低い方の側を中央に向かって移動させます。なぜなら、低い方の側が水量に影響を与えるからです。
解答#
/**
* @param {number[]} height
* @return {number}
*/
var maxArea = function (height) {
let l = 0, r = height.length - 1;
let ans = 0, max = 0;
while (l < r) {
let h = height[l] < height[r] ? height[l] : height[r];
max = (r - l) * h;
height[l] < height[r] ? l++ : r--;
ans = max > ans ? max : ans;
}
return ans;
};