题目地址:
https://leetcode.com/problems/container-with-most-water/
给定一个数组 h h h,找出 i i i和 j j j, i < j i<j i<j,使得 min ( h [ i ] , h [ j ] ) ∗ ( j − i ) \min(h[i], h[j])*(j-i) min(h[i],h[j])∗(j−i)最大,返回那个最大值。
算法的核心是枚举,枚举顺序是,让 i i i从 0 0 0到 l h − 1 l_h-1 lh−1的同时,对每个 i i i,我们从 l h − 1 l_h - 1 lh−1到 i i i来枚举 j j j,只不过提前把不可能的答案排除了。设 f ( i , j ) = min ( h [ i ] , h [ j ] ) ∗ ( j − i ) f(i,j)=\min(h[i], h[j])*(j-i) f(i,j)=min(h[i],h[j])∗(j−i),容易知道,若 h [ i ] < h [ j ] h[i]<h[j] h[i]<h[j],那么 ∀ j ′ < j \forall j'<j ∀j′<j,都有: f ( i , j ) = h [ i ] ∗ ( j − i ) > h [ i ] ∗ ( j ′ − i ) ≥ min ( h [ i ] , h [ j ′ ] ) ∗ ( j ′ − i ) = f ( i , j ′ ) f(i,j)=h[i]*(j-i)>h[i]*(j'-i)\\\ge\min(h[i],h[j']) *(j'-i)=f(i,j') f(i,j)=h[i]∗(j−i)>h[i]∗(j′−i)≥min(h[i],h[j′])∗(j′−i)=f(i,j′)所以此时再枚举 ( i , j ′ ) (i,j') (i,j′)没有意义了,所以直接让 i i i加 1 1 1,也就是枚举左边为 h [ i + 1 ] h[i+1] h[i+1]的可能答案。如此反复,直到枚举完毕。代码如下:
class Solution {
public:
int maxArea(vector<int>& h) {
int n = h.size();
int res = 0;
for (int i = 0, j = n - 1; i < j;) {
res = max(res, min(h[i], h[j]) * (j - i));
if (h[i] < h[j])
i++;
else if (h[i] > h[j])
j--;
else
i++, j--;
}
return res;
}
};
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。