又是一道思路非常简单,但是操作时需要技巧的题。基本思路就是每个元素B[i] = A[1:i-1]的阶乘*A[i+1:]的阶乘。为了避免暴力法求B中每个元素时都要算两遍阶乘,可以事先把A从左到右的阶乘和从右到左的阶乘算出来并储存。
代码如下:
class Solution:
def constructArr(self, a: List[int]) -> List[int]:
left = []
right = []
prod_l = 1
prod_r = 1
b = []
for i in range(len(a)):
prod_l *= a[i]
left.append(prod_l)
prod_r *= a[len(a)-i-1]
right.append(prod_r)
if len(a)<2: return []
b.append(right[-2])
for j in range(1,len(a)-1):
b.append(left[j-1]*right[len(a)-j-2])
b.append(left[-2])
return b
我的这段代码非常不简练,其实right和left两个数组都可以简化去掉, 代码如下:
class Solution:
def constructArr(self, a: List[int]) -> List[int]:
b, tmp = [1] * len(a), 1
for i in range(1, len(a)): #1 to n-1
b[i] = b[i - 1] * a[i - 1] # 左边的阶乘直接储存在b中
for i in range(len(a) - 2, -1, -1): #n-2 to 0
tmp *= a[i + 1] # 右边阶乘
b[i] *= tmp
你以为到这就是巅峰了?那你就太小看leetcode上的大神了,整个过程其实可以合并到一个循环中,比如在同一个循环中可以计算b[1]乘以右边阶乘和b[n-2]乘以左边阶乘,不必拘泥于一个循环中只能指针i所指向的位置:
class Solution:
def constructArr(self, a: List[int]) -> List[int]:
b = [1] * len(a)
left, right = 1, 1
for i in range(1,len(a)):
left *= a[i-1]
b[i] *= left
right *= a[len(a)-i]
b[len(a)-i-1] *= right
return b
不过这种方法虽然简练但是速度不如第二种方法,看来第二种还是巅峰XD