LeetCode 238
题目里面给出了条件:
Note: Please solve it without division and in O(n).
这个增加了难度,否则的话,我们先全部先乘起来,然后除以每个位置上的数就可以o(N)得到答案。当然如果要考虑是否溢出是额外的问题。
如果不能用除法,那么只能想办法乘,如何减少遍历的次数,那么只能考虑做左×和右×,然后再组合。
def productExceptSelf(self, nums: List[int]) -> List[int]:
if nums == None or len(nums) <= 1: return nums
result = [1]
for n in nums:
result.append(result[-1]*n)
result.pop()
rightProduct = 1
for i in range(len(nums)-2, -1, -1):
rightProduct = rightProduct * nums[i+1]
result[i] = result[i] * rightProduct
return result
先把自己左边的都乘起来,然后再遍历一次,把右边的都乘上。
控制使用的space也是o(N),那么我么就重复利用输出结果的数组,直接在上面操作。