码迷,mamicode.com
首页 > 其他好文 > 详细

Maximum Product Subarray

时间:2015-01-24 22:38:58      阅读:254      评论:0      收藏:0      [点我收藏+]

标签:

https://oj.leetcode.com/problems/maximum-product-subarray/

Find the contiguous subarray within an array (containing at least one number) which has the largest product.

For example, given the array [2,3,-2,4],
the contiguous subarray [2,3] has the largest product = 6.

解题思路:

这道题一看就是比较典型的动态规划题目。但和前面的max sum subarray相比,更为巧妙。

令子问题dp[i]为A[i]结尾的数组的最大乘积,状态转移方程就是dp[i] = max(A[i] * dp[i - 1], A[i])。不对,这里发现如果A[i]为负数的话,那么dp[i]就不行了,还必须记录一个A[i-1]结尾的数组的最小乘积,于是我们有第二个dp_min[i]。这样从低至上计算下来。

当然,这里完全可以不考虑A[i]的正负情况,直接dp_max[i] = max(A[i] * dp_max[i - 1], A[i], A[i] * dp_min[i - 1]),dp_min的计算同理。代码更为简洁,时间复杂度不变。

public class Solution {
    public int maxProduct(int[] A) {
        int[] dp_max = new int[A.length];
        int[] dp_min = new int[A.length];
        dp_max[0] = A[0];
        dp_min[0] = A[0];
        
        int max = A[0];
        int min = A[0];
        
        for(int i = 1; i < A.length; i++){
            if(A[i] > 0){
                dp_max[i] = Math.max(A[i] * dp_max[i - 1], A[i]);
                dp_min[i] = Math.min(A[i] * dp_min[i - 1], A[i]);
            }
            if(A[i] < 0){
                dp_max[i] = Math.max(A[i] * dp_min[i - 1], A[i]);
                dp_min[i] = Math.min(A[i] * dp_max[i - 1], A[i]);
            }
            if(A[i] == 0){
                dp_max[i] = 0;
                dp_min[i] = 0;
            }
            if(dp_max[i] > max){
                max = dp_max[i];
            }
            if(dp_min[i] < min){
                min = dp_min[i];
            }
        }
        return max;
    }
}

 

Maximum Product Subarray

标签:

原文地址:http://www.cnblogs.com/NickyYe/p/4246607.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!