构建乘积数组
题目:给定一个数组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]。不能使用除法。
解题思路:先算下三角中的连乘,即我们先算出B[i]中的一部分,然后倒过来按上三角中的分布规律,把另一部分也乘进去。
代码:
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
vector<int> res;
res.push_back(1);
for(int i=1;i<A.size();i++)
res.push_back(res[i-1]*A[i-1]);
int tmp=1;
for(int i=A.size()-2;i>=0;i--)
{
tmp*=A[i+1];
res[i]*=tmp;
}
return res;
}
};