给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
以[1,2,3,4]为例
- 先计算i元素左边的乘积
[1, 1, 2, 6] - 再计算i元素右边的乘积
[24, 12, 4, 1] - 上下同位置同时相乘
[24, 12, 8, 6]
把2,3两步同时操作,结果存在第一步结果中,空间复杂度就是常数了
代码
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
result = [1 for _ in range(len(nums))]
k = 1
for i in range(0, len(nums)):
if i == 0:
result[i] = 1
else:
k *= nums[i - 1]
result[i] = k
k = 1
for i in range(len(nums) - 1, -1, -1):
if i != len(nums) - 1:
k *= nums[i + 1]
result[i] *= k
return result
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/product-of-array-except-self
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。