给你一个整数数组 nums
,返回 数组 answer
,其中 answer[i]
等于 nums
中除 nums[i]
之外其余各元素的乘积 。
题目数据 保证 数组 nums
之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请 不要使用除法,且在 O(n)
时间复杂度内完成此题。
提示:
2 <= nums.length <= 105
-30 <= nums[i] <= 30
- 保证 数组
nums
之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内
解:前后缀相乘,定义两个数组:temp_left[ i ] 表示第 i 个元素的前缀 [0, i - 1] 的乘积,temp_right[ i ] 表示第 i 个元素的后缀 [i + 1, nums.length - 1] 的乘积。
思路:Krahets
class Solution {
public int[] productExceptSelf(int[] nums) {
int[] temp_left = new int[nums.length];
int[] temp_right = new int[nums.length];
int[] ans = new int[nums.length];
temp_left[0] = 1;
temp_right[nums.length - 1] = 1;
for(int i = 1; i < nums.length; i++){
temp_left[i] = nums[i - 1] * temp_left[i - 1];
}
for(int i = nums.length - 2; i >= 0; i--){
temp_right[i] = nums[i + 1] * temp_right[i + 1];
}
for(int i = 0; i < nums.length; i++){
ans[i] = temp_left[i] * temp_right[i];
}
return ans;
}
}
优化:可以看到求后缀乘积的同时就可以获取ans了,求前缀乘积的时候就可以存到ans,之后用tmp变量存储当前后缀乘积即可。
class Solution {
public int[] productExceptSelf(int[] nums) {
int[] ans = new int[nums.length];
ans[0] = 1;
for(int i = 1; i < nums.length; i++) {
ans[i] = ans[i - 1] * nums[i - 1];
}
int tmp = 1; // 存储后缀乘积
for(int i = nums.length - 2; i >= 0 ;i--){
tmp *= nums[i + 1];
ans[i] *= tmp;
}
return ans;
}
}