剑指 Offer 66. 构建乘积数组

Posted 2021-06-26 Blocking The Sky

题目描述

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

示例：

输入: [1,2,3,4,5]
输出: [120,60,40,30,24]

思路

1、非常明确的是，第i位的值为前i-1位的累积再乘以后n-i位的累积
2、先从左到右累积，第i位的值位前i-1位的累积
3、再从右到左累积，第i位的值为2.的结果乘上后n-i位的累积

代码

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