剑指Offer（五十一）：构建乘积数组（C++/Python）

最新推荐文章于 2023-08-16 08:57:52 发布

原创最新推荐文章于 2023-08-16 08:57:52 发布 · 227 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 专栏收录该内容

67 篇文章

订阅专栏

博客围绕数组构建问题展开，给定数组A，要求构建数组B，其中B[i]为A中除A[i]外所有元素的乘积，且不能用除法。解题思路是将B[i]拆分为两部分乘积，分别设为C[i]和D[i]，并给出了C++和Python版代码思路。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定一个数组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[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]看成A[0]*A[1]*...*A[i-1]和A[i+1]*...*A[n-1]两部分的乘积，不妨设C[i] = A[0]*A[1]*...*A[i-1], D[i] = A[i+1]*...*A[n-1]，则C[i] = C[i-1]*A[i-1]，D[i] = D[i+1] * A[i+1]

C++版

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        int len = A.size();
        vector<int>B(len);
        if(len == 0)
            return B;
        B[0] = 1;
        for(int i = 1; i < len; ++ i)
            B[i] = B[i-1] * A[i-1];
        double temp = 1;
        for(int i = len -2; i >= 0; -- i){
            temp *= A[i+1];
            B[i] *= temp;
        }
        return B;
    }
};

Python版

class Solution:
    def multiply(self, A):
        # write code here
        B = []
        if len(A) == 0:
            return B
        B.append(1)
        for i in range(1,len(A)):
            B.append(A[i-1] * B[i-1])
        temp = 1
        for i in range(len(A)-2, -1, -1):
            temp *= A[i+1]
            B[i] *= temp
        return B