LeetCode238_除自身以外数组的乘积

1. 题目

给你一个整数数组 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 <= 105
-30 <= nums[i] <= 30
保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在  32 位 整数范围内
 
进阶：你可以在 O(1) 的额外空间复杂度内完成这个题目吗？（ 出于对空间复杂度分析的目的，输出数组不被视为额外空间。）

2. 题解

2.1 判断0的个数方法

from typing import List


class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        n = len(nums)
        if n <= 1:
            return []

        # 查出0的个数
        count_zero = 0
        data = 1
        res = []

        for i in range(n):
            if nums[i] == 0:
                count_zero += 1
            else:
                data *= nums[i]

        if count_zero >= 2:
            res = [0 for _ in range(n)]

        if count_zero == 1:
            for i in range(n):
                if nums[i] == 0:
                    res.append(data)
                else:
                    res.append(0)

        if count_zero == 0:
            for i in range(n):
                res.append(data // nums[i])

        return res


if __name__ == '__main__':
    s = Solution()
    a = s.productExceptSelf([1,2,3,4])
    print(a)

2.2 暴力法，超时

from typing import List


# 暴力法，不行
class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        n = len(nums)
        if n <= 1:
            return []
        res = []

        for i in range(n):
            temp = 1
            for j in range(n):
                if i != j:
                    temp *= nums[j]
            res.append(temp)
        return res