题目
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
- 输入 保证 数组 answer[i] 在 32 位 整数范围内
进阶:你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组 不被视为 额外空间。)
我的代码(Python)
version 1
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
n = len(nums)
preResult = [nums[0]]
subResult = [nums[-1]]
for i in range(1, n - 1):
preResult.append(preResult[-1] * nums[i])
subResult.append(subResult[-1] * nums[n - i - 1])
answer = [subResult[-1]]
for i in range(1, n - 1):
answer.append(preResult[i - 1] * subResult[n - i - 2])
answer.append(preResult[-1])
return answer
前缀后缀的计算方式。时间复杂度还好,也没用除法,但是仍然有大量重复计算内容,空间复杂度也不是O(1)。
version 2
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
n = len(nums)
answer = [1 for _ in range(n)]
for i in range(1, n):
answer[i] = answer[i - 1] * nums[i - 1]
backTmp = 1
for i in range(n - 1, -1, -1):
answer[i] *= backTmp
backTmp *= nums[i]
return answer
实际上就是前一种算法的优化,空间复杂度降低到了O(1),时间复杂度我觉得没变化,不过却还是击败了61%,比前一版本有很大提升。只是还是觉得有重复计算。
点评
官方的解法和我的思路是一样的。仔细想想,如果要优化计算结果,可能会更麻烦,比如缓存所有的i – j的乘积,以备后续计算,但是不太可能,算法也太复杂。
我想多了。
baddif@gmail.com
AI简历优化站
AI求职跟踪器(建设中)
主站(建设中)
相关平台
Github Issues / Notion / Blog