#陣列元素相乘_Task 30. 陣列:238. 除自身以外陣列的乘積
阿新 • • 發佈:2021-01-23
技術標籤:#陣列元素相乘
Task 30. 陣列:238. 除自身以外陣列的乘積
238. 除自身以外陣列的乘積 - 力扣(LeetCode)leetcode-cn.com題目描述:
給定長度為 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