题目描述:
题号:238
给你一个整数数组 nums
,返回 数组 answer
,其中 answer[i]
等于 nums
中除 nums[i]
之外其余各元素的乘积 。
题目数据 保证 数组 nums
之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请 不要使用除法,且在 O(n)
时间复杂度内完成此题。
示例 1:
输入: nums = [1,2,3,4]
输出:[24,12,8,6]
边界条件:
-
2 <= nums.length <= 105
-
-30 <= nums[i] <= 30
-
保证 数组
nums
之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内
进阶:你可以在 O(1)
的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组 不被视为 额外空间。)
解题思路:
思路一:原地相乘(两步)
算法思路:
1、令 answer 数组的索引 i 对应 i 左侧所有数字的乘积
2、令一个变量 rightSum 储存 i 右侧所有变量的乘积并与 answer[i] 相乘得到结果
时间复杂度:O(N)
空间复杂度:O(1)
C++
// C++
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
if(nums.size() == 0) {
return {};
}
vector<int> answer(nums.size());
// 边界条件
answer[0] = 1;
for(int i = 1; i < nums.size(); i++) {
answer[i] = nums[i - 1] * answer[i - 1];
}
// rightSum 为右侧所有元素的乘积
// 刚开始没有元素,因此边界条件为rightSum = 1
int rightSum = 1;
for(int i = nums.size() - 1; i >= 0; i--) {
// 对于索引 i,左边乘积为 answer[i],右边乘积为 rightSum
answer[i] = answer[i] * rightSum;
// rightSum 是右边所有元素的乘积,所以计算下一个结果需要将当前索引的元素乘到 rightSum 上
rightSum *= nums[i];
}
return answer;
}
};
go
// go
func productExceptSelf(nums []int) []int {
if len(nums) == 0 {
return nil
}
answer := make([]int, len(nums))
// 边界条件
answer[0] = 1
for i := 1; i < len(nums); i++ {
answer[i] = answer[i - 1] * nums[i - 1]
}
// rightSum 为右侧所有元素的乘积
// 刚开始没有元素,因此边界条件为rightSum = 1
rightSum := 1
for i := len(nums) - 1; i >= 0; i-- {
// 对于索引 i,左边乘积为 answer[i],右边乘积为 rightSum
answer[i] = answer[i] * rightSum
// rightSum 是右边所有元素的乘积,所以计算下一个结果需要将当前索引的元素乘到 rightSum 上
rightSum *= nums[i]
}
return answer
}