剑指OFFER 构建乘积数组

剑指OFFER 构建乘积数组

题目描述

给定一个数组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) {
        vector<int> res;
        for(int i=0;i<A.size();i++)
        {
            int e=1;
            for(int j=0;j<A.size();j++)
            {
                if(i==j)continue;
                e *= A[j];
            }
            res.push_back(e);
        }
        return res;
    }
};

代码(去掉重复操作)

看了一些资料才领会题目意图,分两部分处理

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        vector<int> res;
        vector<int> first_half(A.size(),1);
        vector<int> second_half(A.size(),1);
        //上半部分
        for(int i=1;i<A.size();i++)
        {
            first_half[i]=A[i-1]*first_half[i-1];
        }
        //下半部分
        for(int i=A.size()-2;i>=0;i--)
        {
            second_half[i]=A[i+1]*second_half[i+1];
        }
        //整合两部分
        for(int i=0;i<A.size();i++)
        {
            res.push_back(first_half[i]*second_half[i]);
        }
        return res;
    }
};