牛客之构建乘积数组

题目描述
给定一个数组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 len=A.size();
            vector<int>B(len);
            B[0]=1;//初始化乘积数组第一个数字
            /*
将乘积数组分为两部分，第一部分A[0]*A[1]*...*A[i-1]
			*/
            for(int i=1;i<len;i++)
            {
                B[i]=B[i-1]*A[i-1];
            }
            /*
第二部分A[i+1]*...*A[n-1]
			*/
            int temp=1;
            for(int i=len-2;i>=0;i--)
            {
                temp*=A[i+1];
                B[i]*=temp;
            }
            return B;
        }
    };