要求
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
提示: 题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请不要使用除法, 且在 O(n) 时间复杂度内完成此题。
进阶: 你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
核心代码
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
n = len(nums)
res = [1] * n
for i in range(1,n):
res[i] = res[i - 1] * nums[i - 1]
right = 1
for i in range(n-1,-1,-1):
res[i] *= right
right *= nums[i]
return res
另一思路
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
n = len(nums)
res = [1] * n
left = right = 1
for l,r in zip(range(n),range(n-1,-1,-1)):
res[l] *= left
left *= nums[l]
res[r] *= right
right *= nums[r]
return res
重点问题
解题思路:
第一种解法:我们使用左右累计的方式,以[1,2,3,4]为例,我们故意累乘到前一位,不会涉及到本位,一轮循环[1,1,2,6],我们在经历右侧累计,右侧第一个不能包括自己,使用1替换,[24,12,8,6],这种方法很巧妙,值得学习。第二种解法:同样是左右累计,注意zip的使用。