题目
1、分析
- 一种方法是分别连乘 n − 1 n-1 n−1 个数字,得到 B[i],该方式的时间复杂度为 O ( n 2 ) O(n^2) O(n2) ,并不是最好的方法。
- 改用另一种方式,由于 B [ i ] = A [ 0 ] ∗ A [ 1 ] ∗ ⋅ ⋅ ⋅ A [ i − 1 ] ∗ A [ i + 1 ] ∗ ⋅ ⋅ ⋅ A [ n − 1 ] B[i]=A[0]*A[1]*···A[i-1]*A[i+1]*···A[n-1] B[i]=A[0]∗A[1]∗⋅⋅⋅A[i−1]∗A[i+1]∗⋅⋅⋅A[n−1],可以将其看成两部分如下:
- C [ i ] = A [ 0 ] ∗ A [ 1 ] ∗ ⋅ ⋅ ⋅ A [ i − 1 ] C[i]=A[0]*A[1]*···A[i-1] C[i]=A[0]∗A[1]∗⋅⋅⋅A[i−1] 和 D [ i ] = A [ i + 1 ] ∗ ⋅ ⋅ ⋅ A [ n − 1 ] D[i]=A[i+1]*···A[n-1] D[i]=A[i+1]∗⋅⋅⋅A[n−1],则 B [ i ] = C [ i ] ∗ D [ i ] B[i]=C[i]*D[i] B[i]=C[i]∗D[i]。
- 因此可以分别求出 C [ i ] 和 D [ i ] C[i]和D[i] C[i]和D[i] ,然后就可以求出对应的 B [ i ] B[i] B[i] 。
- 可以先将 C [ i ] C[i] C[i]的所有值算出来存放到容器中,然后再倒过来求得最终的值,这样该方法的时间复杂度为 O ( n ) O(n) O(n)
2、代码
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
int n=A.size();
vector<int> res(n);
//求 C[i]=A[0]*A[1]*···A[i-1]
res[0]=1;
for(int i=1;i<n;++i)
{
res[i]=res[i-1]*A[i-1];
}
//求 B[i]=C[i]*D[i]
int temp=1;
for(int j=n-1;j>=0;--j)
{
res[j]*=temp;
temp*=A[j];
}
return res;
}
};