数据结构&算法&人工智能

剑指offer编程题—构建乘积数组

2020-03-17  本文已影响0人  零岁的我

题目描述
给定一个数组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]。不能使用除法。(注意:规定B[0] = A[1] * A[2] * ... * A[n-1],B[n-1] = A[0] * A[1] * ... * A[n-2]

题解思路
对每一个乘积结果B[i]分两次求解,先求解前半部分,再求解后半部分。
1)从左往右求解B[i] = A[0]*A[1]*...*A[i-1]
2)从右往左求解B[i] *= A[i+1]*...*A[n-1]

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        vector<int> res(A.size());
        int p;
        int i;
        for(i=0,p=1;i<A.size();p*=A[i++]) res[i]=p;
        for(i=A.size()-1,p=1;i>=0;p*=A[i--]) res[i]*=p;
        return res;
    }
};
上一篇 下一篇

猜你喜欢

热点阅读