【Lintcode】383. Container With Most Water

题目地址:

https://www.lintcode.com/problem/container-with-most-water/description

给定一个数组 A A A,代表挡板的高度,选定两个挡板,两个挡板中间的区域可以看成一个水的盛具,其容积等于较矮的挡板的高度乘以宽度。要求返回最大容积。

思路是对撞双指针,同时更新得到的容积。每次移动指针的时候,都选择能产生更优解的方式。例如,两个指针分别是 l l l r r r,当 A [ l ] < A [ r ] A[l]<A[r] A[l]<A[r]的时候,这个时候 r r r左移并不会得到更好的解(因为盛具高度没变,但底变窄了),所以此时右移 l l l;同理,当 A [ l ] > A [ r ] A[l]>A[r] A[l]>A[r]的时候要左移 r r r;而当 A [ l ] = A [ r ] A[l]=A[r] A[l]=A[r]的时候则需要同时移动两个指针。代码如下:

public class Solution {
    /**
     * @param heights: a vector of integers
     * @return: an integer
     */
    public int maxArea(int[] heights) {
        // write your code here
        if (heights == null || heights.length == 0) {
            return 0;
        }
        
        int res = 0;
        int l = 0, r = heights.length - 1;
        while (l < r) {
            res = Math.max(res, (r - l) * Math.min(heights[l], heights[r]));
            if (heights[l] > heights[r]) {
                r--;
            } else if (heights[l] < heights[r]) {
                l++;
            } else {
                l++;
                r--;
            }
        }
        
        return res;
    }
}

时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)

算法正确性证明:
数学归纳法。当数组长度为 2 2 2时显然成立。当数组长度为 n n n的时候,由于当 A [ l ] < A [ r ] A[l]<A[r] A[l]<A[r]的时候,左挡板是 l l l的情况下的所有解都不可能超过 A [ l ] ( r − l ) A[l](r-l) A[l](rl),而 A [ l ] ( r − l ) A[l](r-l) A[l](rl)已经更新在res里了,所以不需要枚举左挡板是 l l l的情况了,将 l l l右移一位后,就得到了相同的但规模更小的问题,由归纳假设,算法可以得到正确答案,再和 A [ l ] ( r − l ) A[l](r-l) A[l](rl)比较一下即得到全局最优解。其他情况类似。

证明的本质是,将全局问题分为两类,即以 l l l为左挡板和不以 l l l为左挡板的两类,然后分别的最优解合起来的更优解即为全局最优解。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值