題目出處
238. Product of Array Except Self
難度
medium
個人範例程式碼
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
if not nums:
return []
result = [1 for _ in range(len(nums))]
prefix_product = 1
for i in range(len(nums)):
result[i] *= prefix_product # first *= 1
prefix_product *= nums[i]
postfix_product = 1
for i in range(len(nums)-1, -1, -1):
result[i] *= postfix_product # first *= 1
postfix_product *= nums[i]
return result
算法說明
這題要 O(n^2) 時間做出來非常容易,但題目先是希望要求 O(n) 時間算完,
既然都要求這樣的時間,我們勢必要做一些特殊的處理。
這裡我們用到類似 prefixSum 的概念 (我們用的是 prefixProduct)
因為仔細觀察,我們會發現答案就等於「前面積*後面積」,而數字分布是「前面積、(該數字)、後面積」,
因此我們可以簡單的用一個數字紀錄答案。
同時我們也可以實現「 O(1) 空間」的題目追加要求
input handling
如果沒有 nums, 回傳 [] (題目沒特別要求)
Boundary conditions
用兩次 for 迴圈控制範圍
