AcWing 86. 构建乘积数组

题目描述

给定一个数组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]。

不能使用除法。

样例

输入：[1, 2, 3, 4, 5]

输出：[120, 60, 40, 30, 24]

思考题：

• 能不能只使用常数空间？（除了输出的数组之外）

问题分析 

此题先创建一个ans数组，然后遍历两次给定数组nums，第一次从左往右，第二次从右往左，就得到了结果并返回。第一次遍历是将每个位置前面的乘积保存在当前位置。第二次遍历是将每个位置后面的乘积乘以当前位置之前算出的结果，就得到了该位置的最终结果。

代码实现 

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