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]
.
Follow up:
Could you solve it with constant space complexity? (Note: The output array does not count as extra space for the purpose of space complexity analysis.)
解题思路
两次遍历,第一次遍历,将0
到i-1
的乘积放入res[i]
;第二次遍历,记录i+1
到len-1
的乘积,再与左边的乘积相乘,得到最终结果放入res[i]
。
实现代码
// Runtime: 3 ms
public class Solution {
public int[] productExceptSelf(int[] nums) {
int[] res = new int[nums.length];
Arrays.fill(res, 1);
int left = 1;
for (int i = 0; i < nums.length - 1; i++) {
left *= nums[i];
res[i + 1] = left;
}
int right = 1;
for (int i = nums.length - 1; i > 0; i--) {
right *= nums[i];
res[i - 1] *= right;
}
return res;
}
}