剑指Offer:构建乘积数组(一刷)

来源:互联网 发布:apache kylin 安装 编辑:程序博客网 时间:2024/06/05 19:09

题目描述

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


我的想法:

在纸上画了一个二维矩阵。

为减少循环和运算的次数,使每次计算结果尽量能够得到使用。

声明了两个数组,循环,分别保存上三角和下三角(均不包含对角线上元素)每行元素的乘积结果。

将上、下三角对应元素相乘,得到最终的结果数组。


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





原创粉丝点击