一、题目
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
提示:题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
二、思路——左积乘右积
乘积不包含 nums[i],我们可以分别求出 nums[i] 左边数字的乘积和右边数字的乘积,然后左右相乘,就是最后的结果。
我们可以将 nums[i] 的位置看成 1,如下表。1 将数组分成左右两边,遍历两次数组,第一次求出左积,第二次在左积的基础上乘上右积,得到结果。
res | i = 0 | i = 1 | … | i = n - 2 | i = n - 1 |
---|---|---|---|---|---|
res[0] | 1 | num[1] | … | num[n-2] | num[n-1] |
res[1] | num[0] | 1 | … | num[n-2] | num[n-1] |
… | … | … | … | num[n-2] | num[n-1] |
res[n-2] | num[0] | num[1] | … | 1 | num[n-1] |
res[n-1] | num[0] | num[1] | … | num[n-2] | 1 |
三、代码
class Solution {
public int[] productExceptSelf(int[] nums) {
int[] res = new int[nums.length];
// 左积
int p = 1;
for(int i = 0; i < nums.length; i++){
res[i] = p;
p *= nums[i];
}
// 在左积的基础上,乘上右积
int q = 1;
for(int j = nums.length - 1; j > 0; j--){
q *= nums[j];
res[j - 1] *= q;
}
return res;
}
}