LeetCode 238. Product of Array Except Self(java)

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].

思路:由于每个位置上的值都是除了自己的其他所有数的乘积,然而要求时间复杂度是O(n),证明我们的方法只能是往前走,不能回头,然而走一次并不能得到想要的结果,因此,我们需要走两次,先从前向后走一次,每次记录下除了自己的之前的数的乘积,再从后向前走一次,每次乘以除了自己的后面的数的乘积。这样就可以得到最终想要的结果数组了。

    public int[] productExceptSelf(int[] nums) {
        if (nums == null || nums.length <= 1) return nums;
        int[] helper = new int[nums.length];
        helper[0] = 1;
        for (int i = 1; i < nums.length; i++) {
            helper[i] = helper[i - 1] * nums[i - 1];
        }
        int right = 1;
        for (int i = nums.length - 1; i >= 0; i--) {
            helper[i] = helper[i] * right;
            right = right * nums[i];
        }
        return helper;
    }

Folllow up,我们来想想有没有其他方法,如果要求空间复杂度为O(1),怎么写?

我的思路:先遍历一遍,求出总乘积,然后遍历一遍数组,每个index上为总乘积除以这个位置的值即可。

你可能感兴趣的:(Array)