Leetcode 238题:除自身以外数组的乘积 (python)
注:本人文章中内容多来自于各大佬的经验总结,仅供自身复习及学习,并无商业用途,若有时忘记标注作者及出处还请提醒,敬请见谅。
错误思路记录
我一开始的想法十分简单,既然要求数组乘积和,那么直接使用python内置的reduce()函数就可以了。
刷题菜鸡完全没有想到要使用算法,于是写出了如下代码:
from functools import reduce
nums = list(map(int, input().lstrip('[').rstrip(']').split(',')))
results = [None] * len(nums)
for i in range(len(nums)):
t = nums.pop(i)
results[i] = reduce(lambda x,y:x*y, nums)
nums.insert(i, t)
print(results)
提交之后运行起来我才发现自己的想法过于简单,运行结果报错如下:
然后我才发现靠python内置函数的想法不靠谱,需要设计一种高效的算法。
参考思路
我在leetcode的精选里看到一个高赞回答,觉得很妙,在此将其思路记录下来,以供以后参考:
求除自身以外数组的乘积,可以分两步,先求每个数左侧所有数的乘积,再将其与其右侧所有数相乘,即得所需结果。
根据此思路写出的代码为:
class Solution:
def productExceptSelf(self, nums: List[int]) -> List[int]:
res, p, q = [1], 1, 1
# 每个数左边的数之积
for i in range(len(nums)-1):
p *= nums[i]
res.append(p)
# 再乘以右侧每个数之积
for i in range(len(nums)-1, 0, -1):
q *= nums[i]
res[i-1] *= q
return res
大佬不愧是大佬,思路真的妙,以后还是要多多学习!