LeetCode 238. Product of Array Except Self 题解

LeetCode 238. Product of Array Except Self 题解题目描述给你一个整数数组 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]解题思路使用前缀积和后缀积的思想。answer[i] (nums[0] * ... * nums[i-1]) * (nums[i1] * ... * nums[n-1])先计算前缀积再计算后缀积代码实现def productExceptSelf(nums): n len(nums) answer [1] * n # 计算前缀积 prefix 1 for i in range(n): answer[i] prefix prefix * nums[i] # 计算后缀积并乘到结果中 suffix 1 for i in range(n - 1, -1, -1): answer[i] * suffix suffix * nums[i] return answer复杂度分析时间复杂度O(n)空间复杂度O(1)输出数组不算额外空间总结本题是前缀积技巧的经典应用。关键点分别计算前缀积和后缀积空间优化直接在输出数组中计算不使用除法