题目
给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
思路
这道题最直接的解法就是计算全部的积然后除以自己得到对应的结果(注意0的处理),但题目要求我们用常数空间且不能用除法,所以pass这种思路。
因此我们使用累乘积,输出数组不被实为额外空间,所以我们想要保存的值可以存放在结果数组中。
一共进行两次遍历,一次从输入数组左到右,维护从左至右累乘积放于数组中,然后再从右遍历至左,我们用一个值就可以保存从右至左的累乘积并乘到输出数组对应的值中。
代码
public class problem238 {
public int[] productExceptSelf(int[] nums) {
int[] res=new int[nums.length];
int k=1;
//保存从左到右的累乘积
for(int i=0;i<nums.length;i++){
res[i]=k;
k=k*nums[i];
}
k=1;
//保存从右到左的累乘积,并乘到输出数组中
for(int i=nums.length-1;i>=0;i--){
res[i]=res[i]*k;
k=k*nums[i];
}
return res;
}
public static void main(String[] args) {
problem238 pro=new problem238();
int[] nums={1,2,3,4};
int[] new_nums=pro.productExceptSelf(nums);
for(int i:new_nums){
System.out.print(i+" ");
}
}
}
24 12 8 6