1. 问题描述:
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入:
[1,2,3,4]
输出:
[24,12,8,6]
提示:题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/product-of-array-except-self
2. 思路分析:
分析题目可以知道我们不能在线计算所有元素乘积然后在循环中再除以当前的元素得到当前位置的结果,因为可能在整除的时候可能会出现除数为0的情况,我们可以先计算出列表中每个位置的前缀乘积,因为还需要计算当前位置后面所有元素的乘积所以我们可以考虑从后往前遍历元素,这样在遍历的时候就可以维护当前位置后面所有元素的乘积,而且可以少声明一个列表来存储后缀乘积,当前位置前缀乘积与后缀乘积那么就是当前位置除其他元素之后的元素乘积。
3. 代码如下:
from typing import List
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
# 使用前后缀分解的思路
n = len(nums)
# 前缀列表
pre = [1] * n
for i in range(1, n):
pre[i] = pre[i - 1] * nums[i - 1]
s = 1
for i in range(n - 1, -1, -1):
pre[i] *= s
s *= nums[i]
return pre