算法1
(动归) O(n)
用两个数组left和right,left[i]=A[0]*A[1]*…*A[i-1], left[i]=A[i-1]*left[i-1]; right[i] = A[i+1]*A[i+2]*…*A[n-1],则right[i]=A[i+1]*right[i+1]。
最后结果B[i]=left[i]*right[i]。
时间复杂度分析:需要遍历数组,复杂度为O(n)
C++ 代码
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
vector<int>left(A.size(),1);
vector<int>right(A.size(),1);
for(int i = 1;i<A.size();i++){
left[i] = A[i-1]*left[i-1];
}
for(int i = A.size()-2;i>=0;i--){
right[i] = A[i+1]*right[i+1];
}
vector<int>B(A.size(),0);
for(int i = 0;i<A.size();i++){
B[i] = left[i]*right[i];
}
return B;
}
};
改了一下楼主的代码,这个应该常数了
class Solution { public: vector<int> multiply(const vector<int>& A) { vector<int> B(A.size()+1, 1); for(int i = A.size()-1;i>=0;i--) B[i] = A[i]*B[i+1]; int cum = 1; for(int i = 0;i < A.size();i++){ B[i] = cum*B[i+1]; cum *= A[i]; } B.pop_back(); return B; } };
只能开一个数组,可以用变量存储前一计算结果
思路赞赞赞!