题目描述
给定一个数组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]。不能使用除法
1.直接暴力解法,碰到相应的位置*1,其余位置*A[i],时间复杂度O(n2)
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
vector<int> v(A.size(),1);
for(int i = 0; i < A.size(); ++i)
{
for(int j = 0; j < A.size(); ++j)
{
if(i == j){
v[i]*=1;
}else{
v[i]*=A[j];
}
}
}
return v;
}
};
2.利用二维矩阵的思想,只有中间一条对角线上的数是1,时间复杂度O(N)
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
int length = A.size();
vector<int> v(length,1);
if(length!=0){
v[0] = 1;
for(int i = 1; i < length; i++){
v[i] = v[i-1] * A[i-1];
}
int temp = 1;
for(int i = length - 2; i >= 0; i--){
temp *= A[i+1];
v[i] *= temp;
}
}
return v;
}
};