(C++)剑指offer-51:构建乘积数组
B[i]的意义是A数组不包括i位置的所有乘积,分为i左边的元素乘积和i右边的所有元素乘积。第一个for计算i左边的乘积,第二个for计算右边的。
初始化p=1,是因为0左边没有元素,所以乘积为1。时间复杂度为O(n),具体代码如下:
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
if(A.empty()) return vector<int>();
int n = A.size();
vector<int> B(n);
for(int i = 0, p = 1; i < n; i++){
B[i] = p;
p *= A[i]; //实时更新p,即左边的乘积
}
for(int i = n - 1, p = 1; i >= 0; i--){
B[i] *= p; //乘上右边
p *= A[i]; //实时更新右边的乘积
}
return B;
}
};