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.)
Subscribe to see which companies asked this question
public class Solution {
public int[] productExceptSelf(int[] nums) {
int product = 1;
int zeroCount = 0;
for (Integer item : nums) {
if (item != 0) {
product *= item;
}
if (item == 0) {
++zeroCount;
}
}
int[] outputs = new int[nums.length];
if (zeroCount > 1) {
return outputs;
}
int zeroIndex = 0;
for (int i = 0; i < nums.length; ++i) {
if (zeroCount == 0) {
outputs[i] = product / nums[i];
} else {
if (nums[i] != 0) {
outputs[i] = 0;
} else {
outputs[i] = product;
}
}
}
return outputs;
}
}