1. 程式人生 > 其它 >#陣列元素相乘_Task 30. 陣列:238. 除自身以外陣列的乘積

#陣列元素相乘_Task 30. 陣列:238. 除自身以外陣列的乘積

技術標籤:#陣列元素相乘

Task 30. 陣列:238. 除自身以外陣列的乘積

238. 除自身以外陣列的乘積 - 力扣(LeetCode)​leetcode-cn.com ff6e6f322ead1a916012e8a084c0b1d6.png

題目描述:

給定長度為 n 的整數陣列 nums,其中 n > 1,返回輸出陣列 output ,其中 output[i] 等於 nums 中除 nums[i] 之外其餘各元素的乘積。

示例:

輸入: [1,2,3,4]
輸出: [24,12,8,6]

說明: 請不要使用除法,且在 O(n) 時間複雜度內完成此題。

進階:

你可以在常數空間複雜度內完成這個題目嗎?( 出於對空間複雜度分析的目的,輸出陣列不被視為額外空間。)

解題思路:

提前算好每個位置左邊的乘積和每個位置右邊的乘積,分別需要O(n) 時間複雜度,最後將左乘積和右乘積按照對應位置相乘,即得到需要的output。

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        leftprod = [1]
        for num in nums[:-1]:
            leftprod.append(leftprod[-1] * num)
        rightprod = [1]
        for num in nums[::-1][:-1]:
            rightprod.append(rightprod[-1] * num)
        for i in range(len(nums)):
            leftprod[i] *= rightprod[len(nums) - 1 - i]
        return leftprod