题目地址
给你一个整数数组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 <= 10^5
-30 <= nums[i] <= 30
保证数组 nums之中任意元素的全部前缀元素和后缀的乘积都在32位整数范围内
进阶:你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
官方题解地址
这题官方题解写得挺好的,先理解方法一,再改进成空间复杂度为O(1)的。
方法一主要思想:
从上面的图可以看出,当前位置的结果就是它左部分的乘积再乘以它右部分的乘积。因此需要进行两次遍历,第一次遍历用于求左部分的乘积,第二次遍历在求右部分的乘积的同时,再将最后的计算结果一起求出来。
(具体过程及代码看官方题解)
方法二:改进空间复杂度
思路
由于输出数组不算在空间复杂度内,那么我们可以将 L 或 R 数组用输出数组来计算。先把输出数组当作 L 数组来计算,然后再动态构造 R 数组得到结果。让我们来看看基于这个思想的算法。
(方法一有三个数组:输出数组answer、左数组L、右数组R。
方法二只用一个数组:输出数组answer,该数组同时充当左数组,而右数组:是在左数组构建完成后,再该数组的基础上动态构建右数组。)
算法
1.初始化 answer 数组,对于给定索引 i,answer[i] 代表的是 i 左侧所有数字的积。
2.构造方式与之前相同,只是我们试图节省空间,先把 answer 作为方法一的 L 数组。
3.这种方法的唯一变化就是我们没有构造 R 数组。而是用一个遍历来跟踪右边元素的乘积。并更新数组answer[i]=answer[i]R。然后R更新为R=Rnums[i],其中变量R表示的就是索引右侧数字的乘积。
代码实现:
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
length = len(nums)
answer = [0]*length
# answer[i] 表示索引 i 左侧所有元素的乘积
# 因为索引为 '0' 的元素左侧没有元素, 所以 answer[0] = 1
answer[0] = 1
for i in range(1, length):
answer[i] = nums[i - 1] * answer[i - 1]
# R 为右侧所有元素的乘积
# 刚开始右边没有元素,所以 R = 1
R = 1;
for i in reversed(range(length)):
# 对于索引 i,左边的乘积为 answer[i],右边的乘积为 R
answer[i] = answer[i] * R
# R 需要包含右边所有的乘积,所以计算下一个结果时需要将当前值乘到 R 上
R *= nums[i]
return answer