Leetcode 238. 除自身以外数组的乘积

238. 除自身以外数组的乘积

一、题目描述

给你一个整数数组 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 位 整数范围内

二、我的想法

我的想法直接在题目上就被否了，不能暴力输出也不能用除法。
又是看的Krahets的题解。238. 除自身以外数组的乘积（清晰图解）
大致就是将数组长度为5的数组，写成5*5的矩阵。以主对角线分界，先计算下三角的各行乘积，再计算上三角的。将两者相乘得到最后结果。

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        numsLen = len(nums)
        newList = [1] * numsLen
        downTotal = 1
        for i in range(0,numsLen-1):
            downTotal *= nums[i]
            newList[i+1] *= downTotal
        upTotal = 1
        for i in range(numsLen - 1, 0, -1):
            upTotal = upTotal * nums[i]
            newList[i-1] *= upTotal
        return newList

再去掉没用的变量，最后大概如下。

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        numsLen = len(nums)
        newList = [1] * numsLen
        tmp = 1
        for i in range(0,numsLen-1):
            newList[i+1] = newList[i] *nums[i]
        for i in range(numsLen - 1, 0, -1):
            tmp *= nums[i]
            newList[i-1] *= tmp
        return newList