题目:构建乘积数组
给定一个数组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]。不能使用除法。
时间复杂度为O(n)的解法
思路:
不妨定义C[i]=A[0]A[1]…A[i-1], D[i]=A[i+1]…*A[n-2]*A[n-1].
C[i]可以用自上而下的顺序计算出来,即C[i]=C[i-1]*A[i-1].
类似的,D[i]可以用自下而上的顺序计算出来,即D[i]=D[i+1]*A[i+1]
来自《剑指offer—名企面试官精讲典型编程题》
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
int length1 = A.size();
vector<int> B(length1);
if( length1 == 0)
return B;
B[0]=1;
for(int i = 1; i < length1; ++i )
B[i]=B[i-1]*A[i-1];
int tmp = 1;
for(int i=length1-2; i>=0; --i){
tmp *= A[i+1];
B[i]*=tmp;
}
return B;
}
};
更简洁的一种写法,来自牛客网LOL
//B[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]
//从左到右算 B[i]=A[0]*A[1]*...*A[i-1]
//从右到左算B[i]*=A[i+1]*...*A[n-1]
//其中左边部分从上到下依次累乘,右边部分从下到上依次累乘
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
int n=A.size();
vector<int> b(n);
int ret=1;
for(int i=0;i<n;ret*=A[i++]){
b[i]=ret;
}
ret=1;
for(int i=n-1;i>=0;ret*=A[i--]){
b[i]*=ret;
}
return b;
}
};