构建乘积数组

构建乘积数组

题目描述

给定一个数组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]。不能使用除法。

解题思路

• 构建两个数组，分别表示A[i]前面的元素的乘积和A[i]后面的元素的乘积
• 乘积数组中位置i的值，就是A[i]前面元素的乘积与A[i]后面元素的乘积相乘

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