最大乘积子阵列的范围(Kadane算法变体)

时间:2014-05-26 04:46:18

标签: java algorithm sub-array kadanes-algorithm

我一直试图获得一个子阵列的最大产品范围(为求职面试而学习)。

这里已经提到过(但没有提供有效答案)。 Getting range of max product subarray using Kadanes algorithm

这里解释了技巧/算法:http://www.geeksforgeeks.org/maximum-product-subarray/

我能够轻松获得最大的产品,但经过多次尝试,仍然无法弄清楚如何获得范围(左右索引正确)。有人可以帮忙吗?

我已粘贴了我的代码,因此您可以快速复制并运行它。

import java.util.*;

public class ArrayMax {

// maximum product
public static int[] getMaxProduct(int[] list)
{
    int max = 1, min = 1, maxProd = 0;
    int l = 0, left = 0, right = 0;

    for (int i = 0; i < list.length; i++) {

        // positive number!
        if (list[i] > 0) {
            max = max * list[i];
            min = Math.min(1, min * list[i]);
        }
        else if (list[i] == 0) {
            max = 1;    // reset all
            min = 1;
            l = i + 1;
        }
        else {
            // hold the current Max                                  
            int tempMax = max;
                     // need to update left here (but how??)
            max = Math.max(min * list[i], 1); // [-33, 3]
            min = tempMax * list[i];  // update min with prev max

        }

    //  System.out.printf("[%d %d]%n", max, min);       
        if (max >= maxProd) {
            maxProd = max;
            right = i;
            left = l;
        }
    }

    System.out.println("Max: " + maxProd);
    // copy array
    return Arrays.copyOfRange(list, left, right + 1);
}


// prints array
public static void printArray(int[] list) {

    System.out.print("[");
    for (int i = 0; i < list.length; i++) {     
        String sep = (i < list.length - 1) ? "," : "";
        System.out.printf("%d%s", list[i], sep);
    }

    System.out.print("]");
}

public static void main(String[] args) {

    int[][] list = {
        {5, 1, -3, -8},
        {0, 0, -11, -2, -3, 5},
        {2, 1, -2, 9}
    };

    for (int i = 0; i < list.length; i++) {
        int[] res = getMaxProduct(list[i]);

        printArray(list[i]);
        System.out.print(" => ");
        printArray(res);

        System.out.println();
    }
}
} 

以下是示例输出:

Max: 120
[5,1,-3,-8] => [5,1,-3,-8]
Max: 30
[0,0,-11,-2,-3,5] => [-11,-2,-3,5]
Max: 9
[2,1,-2,9] => [2,1,-2,9]

正如您所看到的,我获得了最大的产品,但范围是错误的。

Case#2, Max is 30 (correct answer: [-2,-3,5], showing: [-11,-2,-3,5])
Case#3, Max is 9 (correct answer: [9], giving: [2,1,-2,9])

请帮忙。

2 个答案:

答案 0 :(得分:2)

更简单的方法是在计算maxProd(最后)时尝试找到左侧位置/标记。你的正确位置是准确的,所以从左到右设置并按列表[左]划分maxProd,直到你达到1,同时向左递减。那是你到达左边的时候。

返回之前的以下代码应解决它。

int temp = maxProd;
left = right;
while (temp != 1) {
   temp = temp / list[left--];
}
left++;
// copy array
return Arrays.copyOfRange(list, left, right + 1);

答案 1 :(得分:1)

我认为您需要跟踪l的2个值。一个代表数字子阵列的起始索引,乘以最大值,而另一个代表数字子阵列的起始索引,乘以最小值。

然而,更简单的方法是等到你找到最大答案(在maxProd中)及其位置(在右边)。此时,您可以循环遍历数组乘以列表的元素,直到您的总数达到maxProd(从右侧开始并向后迭代)。您乘以的最后一个元素必须是子数组的开头。