给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
思路一:用了除法,其实不满足题意,注意要考虑0的情况,否则会出现0/0的情况,特判一下0的情况即可。
class Solution {
public int[] productExceptSelf(int[] nums) {
int ans[] = new int [nums.length];
int sum = 1;
int zero = 0;
for(int i=0;i<nums.length;i++)
{
if(nums[i]==0)
zero++;
}
if(zero>=2)
return ans;
for(int i=0;i<nums.length;i++)
{
if(nums[i]!=0)
sum = sum*nums[i];
}
if(zero==1)
{
int dir = 0;
for(int i=0;i<nums.length;i++)
{
if(nums[i]==0)
{
dir = i;
break;
}
}
ans[dir] = sum;
return ans;
}
for(int i=0;i<nums.length;i++)
ans[i] = sum/nums[i];
return ans;
}
}
思路二:保存从左到右的乘积和,在保存从右到左的乘积和,这个思路挺不错。
class Solution {
public int[] productExceptSelf(int[] nums) {
int ans[] = new int [nums.length];
int sum = 1;
for(int i=0;i<nums.length;i++)
{
ans[i] = sum;
sum = sum*nums[i];
}
sum = 1;
for(int i=nums.length-1;i>=0;i--)
{
ans[i] = ans[i]*sum;
sum = sum*nums[i];
}
return ans;
}
}