一、题目
给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请不要使用除法,且在 O(n) 时间复杂度内完成此题。
示例 1:
输入: nums = [1,2,3,4] 输出: [24,12,8,6]
示例 2:
输入: nums = [-1,1,0,-3,3] 输出: [0,0,9,0,0]
提示:
- 2 <= nums.length <= 10^5
- -30 <= nums[i] <= 30
- 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内
二、代码
class Solution {
public int[] productExceptSelf(int[] nums) {
int n = nums.length;
// pos[i]表示i位置右侧的(不包括i位置)的后缀乘积
int[] pos = new int[n];
// 记录当前的后缀成绩
int posProduct = 1;
// 生成后缀乘积数组
for (int i = n - 1; i >= 0; i--) {
pos[i] = posProduct;
posProduct *= nums[i];
}
// 记录最后的答案,存储每个位置除自身以外的乘积
int[] ans = new int[n];
// 0下标的结果一定是0下标右侧的后缀乘积
ans[0] = pos[0];
// 记录当前遍历到位置的左侧的前缀成绩(不包括当前位置)
int preProduct = nums[0];
// 从左往右开始遍历求每个位置的除自身以外的乘积
for (int i = 1; i < n; i++) {
// i位置的除自身以外的乘积就等于i左侧的前缀乘积 * i右侧的后缀成绩(因为这两个乘积都包含了除了i位置外其他所有数的乘积了)
ans[i] = preProduct * pos[i];
// 维持preProduct前缀乘积
preProduct *= nums[i];
}
// 返回答案
return ans;
}
}
三、解题思路
题目要求不能用除法,也就不能用将数组所有的数相乘得到总的乘积,然后求每一个位置的时候用总乘积除以这个位置的数来得到结果的方法了。
第一步,先求后缀乘积,将每一个位置后面的后缀乘积求出来,如上图的ans所示,对arr求后缀乘积。
然后在从左往右遍历数组,设置一个变量,专门用来记录当前遍历到的位置左侧的前缀乘积。
当前遍历到的位置的除自身以外的乘积,就是该位置左侧的前缀乘积 * 在前面已经求出来的该位置右边的后缀乘积。
当来到i位置的时候,左边用一个变量维持i位置左侧的前缀乘积,i位置右遍的后缀乘积由之前求出来的后缀乘积数组可以得到。这样遍历一遍我们就求出了最后的答案。