构建乘积数组

给定一个数组 A[0,1,…,n-1],请构建一个数组 B[0,1,…,n-1],其中 B[i] 的值是数组 A 中除了下标 i 以外的元素的积, 即 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]可以看作,B[i]=C[i]×D[i],C[i]=A[0]×A[1]×…×A[i-1],D[i]=A[i+1]×…×A[n-1]
可以先求出C和D,再求答案。

代码:

class Solution {
public:
    vector<int> constructArr(vector<int>& a) {
        int len = a.size();
        if(len == 0) return {};
        vector<int> left(len);
        vector<int> right(len);
        left[0] = 1;
        for(int i = 1;i < len;i++) {
            left[i] = a[i-1]*left[i-1];
        }
        right[len - 1] = 1;
        for(int i = len - 2;i >= 0;i--) {
            right[i] = a[i+1]*right[i+1];
        }
        vector<int> ans(len);
        for(int i = 0;i < len;i++) {
            ans[i] = left[i]*right[i];
        }
        return ans;
    }
};