问题
思路
这个题目还是有一点难度的,很难考虑到。这个方法我也是参考了别人的方法。
对于,B[i]=A[0]A[1]…A[i-1]*A[i+1]…*A[n-1]
可以分成两部分:A[0]A[1]…A[i-1] 以及 A[i+1]A[i+2]…A[n-1]这两部分,而这两部分分别是连续的。
所以,
记 P[i] = A[0]A[1]…A[i-1] = P[i-1] * A[i-1]
同理,Q[i] = A[i+1] * A[i+2] * … A[n-1] = Q[i+1]*A[i+1]
这两个数组可以累乘计算。
最后,B[i] = P[i] * Q[i] .
代码
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
int sz = A.size();
vector<int> ret(sz);
if(!sz) return ret;
vector<int> P(sz); // P[i] = A[0]*A[1]*...A[i-1]
vector<int> Q(sz); // Q[i] = Q[i+1]*Q[i+2]...*Q[n-1]
/* 由于PQ两个数组分别是左连续和右连续,因此可以累乘获得 */
P[0] = 1;
for(int i = 1; i < sz;++i){
P[i] = P[i-1] * A[i-1];
}
Q[sz-1] = 1;
for(int i = sz-2; i >= 0; --i){
Q[i] = Q[i+1] * A[i+1];
}
for(int i = 0; i < sz; ++i){
ret[i] = P[i] * Q[i];
}
return ret;
}
};