剑指 Offer 66. 构建乘积数组

思路

根据表格的主对角线（全为 1），可将表格分为 上三角 和 下三角 两部分。分别迭代计算下三角和上三角两部分的乘积，即可获得结果。

作者：jyd
链接：https://leetcode-cn.com/problems/gou-jian-cheng-ji-shu-zu-lcof/solution/mian-shi-ti-66-gou-jian-cheng-ji-shu-zu-biao-ge-fe/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

class Solution {
public:
    vector<int> constructArr(vector<int>& a) {
        int n = a.size();
        vector<int> b(n);

        if(n==0) return b;

        // 左边，下三角，自上而下 bi[i]
        b[0] = 1;        
        for(int i=1; i<n;i++){     // 从1开始，b[0]的左边肯定值为1,因为它左边没数了
            b[i] = b[i-1] * a[i-1];
        }  

        // 右边，上三角，自下而上
        // 之所以自下而上，是因为上面正好计算到了b[n-1]
        int temp = 1;
        for(int i=n-1; i>=0;i--){
            b[i] = b[i] * temp;   // b[i]最后一个数右边没有了，乘以1，a[i]留给b[i-1]
            temp *= a[i];    
        }

        return b;
    }
};