题目描述
给定一个数组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];)
思路:列表B中的元素B[i]就是A列表去掉第i个元素后连乘,所以把列表A以i元素为分界分为两个列表,分别求乘积,再乘在一起即为B的i元素
代码如下:
# -*- coding:utf-8 -*-
import numpy
class Solution:
def multiply(self, A):
B = []
for i in range(len(A)):
A1 = A[:i]
A2 = A[i+1:]
B.append(numpy.prod(A1)*numpy.prod(A2))
return B
???在本地运行可以出结果但是牛客上就不可以???numpy不能用么???
方法二:
把第i个元素看做1
import copy
class Solution:
def multiply(self, A):
B = []
for i in range(len(A)):
temp = copy.copy(A)
temp[i]=1
b = 1
for j in temp:
b *= j
B.append(b)
return B