【 声明:版权所有,转载请标明出处,请勿用于商业用途。 联系信箱:libin493073668@sina.com】
题目描述
给定一个数组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]。不能使用除法。
思路
我们开一个数组用来存对于第i位而言,前i-1的乘积。
那么对于新数组ans[i],我们已经求得了A[1]*A[2]*...*A[i-1],
然后我们再从尾到头扫描一遍,对于当前第i位,我们任然按照前面的思想,使用一个变量来累乘A[len-1]*...A[i+1],并更新结果数组
class Solution
{
public:
vector<int> multiply(const vector<int>& A)
{
vector<int> ans;
int len = A.size();
if(len==0)
return ans;
ans.push_back(1);
for(int i = 0; i<len-1; i++)
ans.push_back(ans.back()*A[i]);
int tmp = 1;
for(int i = len-1; i>=0; i--)
{
ans[i]*=tmp;
tmp*=A[i];
}
return ans;
}
};