LeetCode之蓄水量最大的容器(Container With Most Water)

问题描述:
给定n个非负的整数 a1, a2, ..., an , 其中每一个代表坐标系 (i, ai)中的一个点. 画出n条竖线,每条线的起点和终点分别为 (i, ai) 和 (i, 0). 找到其中的两条线,它们和x轴一起组成一个容器使得该容器装水最多。

PS: 容器不会倾斜,并且n最小是2。


上图中的竖线由数组 [1,8,6,2,5,4,8,3,7]给出。 在这个实例中,最大蓄水量容器 (蓝色区域)可蓄水容量是49 .png

示例:

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

一般解法

最容易想到的便是暴力法,针对每条竖线计算其和其前方所有竖线组成的容器蓄水量,最终得到这些竖线组成的所有容器中的最大值。
时间复杂度
计算次数为\sum_{i=1}^ni= {n(n-1)\over 2},从而时间复杂度为O(n^2)
代码实现

private static int solutionV1(int[] height) {
    if (height == null || height.length <= 1) {
        return 0;
    }
    int maxArea = Integer.MIN_VALUE;
    int curArea;
    for (int i = 1; i < height.length; i++) {
        for (int j = i - 1; j >= 0; j--) {
            curArea = Math.min(height[i], height[j]) * (i - j);
            if (maxArea < curArea) {
                maxArea = curArea;
            }
        }
    }
    return maxArea;
}

进阶解法

一般解法的时间复杂度为O(n^2),如果要优化的话,我们需要找到更低阶的解法,比如O(n)
通过观察可以发现,对于任意的i,j(0 =< i < k < j< n),如果h_i<=h_j,那么有:
area(i, j)=min(h_i, h_j) * (j - i)=h_i*(j-i)>h_i*(j-k)>=min(h_i, h_k)(k-i)=area(i,k)
所以对任意的k,area(i,j)>area(i,k),也就是说在所有包含i的容器中最大的就是area(i,j)
area(i,j)=max_{i,j},那么maxArea=max(area(i+1,j),max_{i,j})
同理,如果h_i>h_j,那么就有:
area(i,j)=max_{i,j},那么maxArea=max(area(i,j-1),max_{i,j})
从上述的推理我们可以看出,如果h_i<=h_j,我们便可以直接把i右移,否则把j左移,从而去掉一些不必要的计算,这样便可以在O(n)时间内得到结果。
代码实现

private static int solutionV2(int[] height) {
    if (height == null || height.length <= 1) {
        return 0;
    }
    int maxArea = 0;
    int curArea;
    int s = 0, e = height.length - 1;
    while (s < e) {
        curArea = height[s] < height[e] ? (e - s) * height[s] : (e - s) * height[e];
        maxArea = maxArea > curArea ? maxArea : curArea;
        if (height[s] < height[e]) {
            s++;
        } else {
            e--;
        }
    }
    return maxArea;
}

总结

该题是典型的需要通过数学推理才能找到更优解法的题型,十分有趣,特此整理纪念。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 很久之前喜欢在豆瓣上写日记、影评,慢慢就懒了,不会好好写完一天的心情,试试看吧,重新记录自己的生活。 回国三个月,...
    Esme55阅读 195评论 0 0
  • 宋仲基上海粉丝见面会亲笔信内容 这是最后一封的信了! 大家好我是仲基,这些日子过得好吗?现在为大家写这封信的时间7...
    歆曼阅读 391评论 5 3
  • 近来读孙郁散文集《秋夜闲谈》,里面有一段摘自废名的议论,颇值得深思: “ 中国文章里没有外国人的厌世观。中国人生在...
    繁露阅读 234评论 0 0
  • export JAVA_7_HOME=/Library/Java/JavaVirtualMachines/jdk1...
    志壹阅读 248评论 0 0
  • 前阵子看《改变自己》微信公众号里推送的一篇文章,这个公众号是我每天必看的,还注册了会员,通过这个会员也对自己的某些...
    猫饼干阅读 171评论 9 0