构建乘积数组

题目描述

给定一个数组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]。不能使用除法。（注意：规定B[0] = A[1] * A[2] * ... * A[n-1]，B[n-1] = A[0] * A[1] * ... * A[n-2];）

对于A长度为1的情况，B无意义，故而无法构建，因此该情况不会存在。

分析

A = [1 2 3 4 5]
n= 5
B[0] = 1*2*3*4*5 = 120
B[1] = 1*3*4*5* = 60
B[2] = 1*2*4*5 = 40
B[3] = 1*2*3*5 = 30
B[4] = 1*2*3*4 = 24

明显可以看出规律来，构建B 中的第一个数是A中所有的数的累积，第二个是A中去掉第二个数，剩下的数的累积。。。。

代码

# -*- coding:utf-8 -*-
import copy
class Solution:
    def multiply(self, A):
        B=[]
        sum=1
        for A_index in range(len(A)):
            C = copy.deepcopy(A)
            if A_index == 0:
                for i in A:
                    sum*=i
                B.append(sum)
                sum=1
            else:
                C.remove(C[A_index])
                for i in C:
                    sum*=i
                B.append(sum)
                sum=1
        return B

if __name__ == "__main__":
    s = Solution()
    result = s.multiply([1,2,3,4,5])
    print(result)