给定长度为 n 的整数数组
nums
,其中 n > 1,返回输出数组output
,其中output[i]
等于nums
中除nums[i]
之外其余各元素的乘积。示例:
输入:[1,2,3,4]
输出:[24,12,8,6]
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
思路:采用后乘积数组或前乘积数组。
从后向前遍历,同时累乘,得到后乘积数组,数组保存的是不包括当前元素的其后所有元素的乘积
从前向后遍历,同时累乘,得到前乘积数组,数组保存的是不包括当前元素的其前所有元素的乘积
为了减少额外空间,只使用后乘数组,使用一个int t保存叠成结果,那么t中保存的是不包括当前元素的前面元素的乘积结果
从前向后遍历后乘数组同时乘以t得到的就是不包括当前元素的前后所有元素乘积。
public int[] productExceptSelf(int[] nums) {
//使用O(n)空间
/*
int[] post = new int[nums.length];
int[] pre = new int[nums.length];
post[nums.length-1]=1;
pre[0]=1;
for(int i=nums.length-2;i>=0;i--){
post[i]=nums[i+1]*post[i+1];//计算后乘数组
}
for(int i=1;i<nums.length;i++){
pre[i]=nums[i-1]*pre[i-1];//计算前乘数组
}
for(int i=0;i<nums.length;i++){
nums[i]=post[i]*pre[i];
}
return nums;*/
//使用O(1)空间
int[] post = new int[nums.length];
int t=1;//额外空间
for(int i=nums.length-1;i>=0;i--){
post[i]=t;
t=t*nums[i];
}
t=1;
for(int i=0;i<nums.length;i++){
post[i]*=t;
t=t*nums[i];
}
return post;
}