238. 除自身以外数组的乘积
mid
给你一个整数数组 nums，返回 数组 answer ，其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请**不要使用除法，**且在 O(n) 时间复杂度内完成此题。
示例 1:
1
2
| 输入: nums = [1,2,3,4]
输出: [24,12,8,6]
|
示例 2:
1
2
| 输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]
|
左右乘积列表
乘积 = 当前数左边的乘积 * 当前数右边的乘积
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
| func productExceptSelf(nums []int) []int {
l, r, res := make([]int, len(nums)), make([]int, len(nums)), make([]int, len(nums))
// l[i]为i左边元素的乘积
// r[i]为i右边元素的乘积
l[0] = 1
for i := 1; i < len(nums); i++ {
l[i] = l[i-1] * nums[i-1]
}
r[len(nums)-1] = 1
for i := len(nums) - 2; i >= 0; i--{
r[i] = r[i+1] * nums[i+1]
}
for i := 0; i < len(nums); i++ {
res[i] = l[i] * r[i]
}
return res
}
|
O(1) 空间解决
用 res 数组来暂时替代 l 数组，在复制的过程中，动态地修改 r[i] 的值（用一个临时变量 r 来表示）
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| func productExceptSelf(nums []int) []int {
res := make([]int, len(nums))
// res[i]暂时为i左边元素的乘积
res[0] = 1
for i := 1; i < len(nums); i++ {
res[i] = res[i-1] * nums[i-1]
}
r := 1 // i右侧元素乘积
for i := len(nums) - 1; i >= 0; i-- {
res[i] = res[i] * r
r *= nums[i]
}
return res
}
|