剑指offer10构建乘积数组

原创于 2020-05-06 09:27:44 发布 · 194 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种不使用除法构建特殊数组B的方法，通过将数组A分成两部分并计算各自的乘积来得到B[i]。提供了两种实现思路，一种利用numpy库计算乘积，但在特定平台上可能受限；另一种通过复制数组并设置目标元素为1来遍历计算。

题目描述
给定一个数组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