arrays - 在 Java 8 中迭代 Java 数组

Question

这是一个简单的算法问题，

Given an array of n integers where n > 1, nums, return an array output such that output[i] is equal to the product of all the elements of nums except nums[i]. Solve it without division and in O(n). For example, given [1,2,3,4], return [24,12,8,6].

这是我的解决方案

public static int[] productExceptSelf(int[] nums) {

    int[] result = new int[nums.length];

    int leftProduct = 1;
    int rightProduct = 1;

    for(int i = 0; i < result.length; i++){
        result[i] = leftProduct;
        leftProduct *= nums[i];
    }

    for(int i=nums.length -1; i >= 0; i --){
        result[i] *= rightProduct;
        rightProduct *= nums[i];
    }
    return result;
}

public static void main(String[] args) {
    int[] output = productExceptSelf(new int[]{1, 2, 3, 4});
    Arrays.stream(output).forEach(System.out::println);
}

这很好用。我想要学习的是如何在 Java 8 中重写此代码。就像 Java 8 中循环的不同选项一样。

Answer 1

您可以用几行稍微不同的代码来完成:

public static int[] productExceptSelf(int[] nums) {
    int all = Arrays.stream(nums).reduce(1, (x, y) -> x * y);
    return IntStream.range(0, nums.length).map(x -> all / nums[x]).toArray();
}