题目
238. 除自身以外数组的乘积
给你一个整数数组 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 <= 105
-30 <= nums[i] <= 30
保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内
进阶:你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组 不被视为 额外空间。)
解题
题目不让用的暴力解法(就除就除),分0的数量讨论,时间O(n)
func productExceptSelf(nums []int) []int {
flagMap := map[int]struct{}{}
for i, v := range nums{
if v == 0{
flagMap[i] = struct{}{}
}
}
if len(flagMap)>1 {
for i := range nums{
nums[i]=0
}
}else if len(flagMap)==1 {
flag := 0
for key := range flagMap{
flag = key
}
nums[flag] = 1
sum := 1
for i:= 0;i<len(nums);i++{
sum *= nums[i]
nums[i]=0
}
nums[flag] = sum
} else {
sum := 1
for i := range nums{
sum*=nums[i]
}
for i:=0 ; i<len(nums);i++{
nums[i] = sum/nums[i]
}
}
return nums
}
题解
方法一:左右乘积列表
1
func productExceptSelf(nums []int) []int {
leftSlice, rightSlice := make([]int, len(nums)), make([]int, len(nums))
leftSlice[0], rightSlice[len(nums)-1] = 1, 1
for i:=1; i<len(nums);i++{
leftSlice[i]= leftSlice[i-1]*nums[i-1]
}
for i:=len(nums)-2; i>=0;i--{
rightSlice[i]= rightSlice[i+1]*nums[i+1]
}
for i:=0; i<len(nums);i++{
nums[i]= leftSlice[i]*rightSlice[i]
}
return nums
}
时间:O(n)
空间:O(n)
方法二:优化方法1
由于输出不算做空间复杂度中,所以可以进一步优化空间复杂度。
func productExceptSelf(nums []int) []int {
ans := make([]int , len(nums))
ans[0] = 1
for i:=1; i<len(nums);i++{
ans[i]= ans[i-1]*nums[i-1]
}
rightCount := nums[len(nums)-1]
for i:=len(nums)-2; i>=0;i--{
ans[i] = ans[i]*rightCount
rightCount *= nums[i]
}
return ans
}
时间:O(n)
空间:O(1)
11111
长进
- 左右状态列表第二次做到了,应该记住此方法了。
- 切片已知cap应该使用make进行创建