# -*- coding: utf-8 -*-
# @Time : 2019-10-07 0:34
# @Author : Jayce Wong
# @ProjectName : job
# @FileName : multiply.py
# @Blog : http://blog.51cto.com/jayce1111
# @Github : https://github.com/SysuJayce
class Solution:
"""
给定一个数组A[0,1,...,n-1],请构建一个数组B[0,1,...,n-1],
其中B中的元素B[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]。不能使用除法。
"""
def multiply(self, A):
"""
解法1:
如果没有规定不能用除法的话,可以按照B的定义去计算每一个元素的值,需要注意除0的异常
但是这种方法的时间复杂度是O(n^2),空间复杂度是O(1)
解法2:
按照B的定义,B[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]
=C[i] * D[i]
C[0] = 1,
C[1] = A[0],
C[2] = A[0] * A[1]
...
C[n-1] = A[0] * A[1] * ... * A[n - 2]
D[0] = A[1] * A[2] * A[3] * ... * A[n - 1]
D[1] = A[2] * A[3] * ... * A[n - 1]
D[2] = A[3] * ... * A[n - 1]
...
D[n - 2] = A[n - 1]
D[n - 1] = 1
由以上的规律可以看出,对于C[i],我们可以自顶向下求解,
对于D[i],我们可以自底向上求解
也就是说,我们可以先用一个和输入A等大的数组来保存C的值,然后自底向上求解D,
也就顺便求出了B。
可以说是利用空间换时间,时间复杂度为O(2*n) = O(n),空间复杂度为O(n)
:param A: 数组
:return: 乘积数组B
"""
C = A[:]
B = A[:]
C[0] = 1
for i in range(1, len(A)):
C[i] = C[i - 1] * A[i - 1]
D = 1
B[-1] = C[-1]
for i in range(len(A) - 2, -1, -1):
D *= A[i + 1]
B[i] = C[i] * D
return B
剑指offer:构建乘积数组
最新推荐文章于 2022-03-02 22:28:28 发布