数组------构建乘积数组

题目描述:
给定数组A[n] = {0, 1, …, n-1},要求构建一个数组B[n],使得B[i] = A[0]A[1]…A[i-1]A[i+1]…A[n-1].并且不能使用除法.
分析:
若能使用除法,则B[i] = A[0]A[1]…A[n-1] / A[i].但由于不能使用除法,所以此方法不可行.
方法一: 可以使用蛮力方法,对每个B[i]都重新乘一遍,从而得到结果,复杂度是O(n^2).
方法二: 令B[i] = C[i] * D[i], 其中C[i] = A[0] * A[1] * … *A[i-1] = C[i-1] *A[i-1]. (C[0] = 1)
D[i] = A[i+1] * A[i+2] * … * A[n-1] = D[i+1] * A[i+1]. (D[n-1] = 1)

vector<int> func(vector<int>& A){
    vector<int> ret;
    ret.push_back(1);
    int num = 1;
    for(int i = 0; i <= A.size()-1; i++){
        num *= A[i-1];
        ret.push_back(num);
    }
    num = 1;
    for(int i = A.size()-2; i >= 0; i--){
        num *= A[i+1];
        ret[i] *= num;
    }
    return ret;
}