剑指offer_牛客网_构建乘积数组 2020-2-23

2020-2-23

链接：剑指offer_牛客网_构建乘积数组
题目描述：
给定一个数组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];）
思路：
时间复杂度为O(N)O(N)
可以把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-2]A[n-1]两部分的乘积。
即通过A[i]项将B[i]分为两部分的乘积。效果相当于是个对角矩阵。

第一个for循环用来计算上图1范围的数，第二个for循环用来计算上图2范围的数。

代码：

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